diff --git a/ggml.c b/ggml.c index cde938eae..9894e13c5 100644 --- a/ggml.c +++ b/ggml.c @@ -5863,7 +5863,7 @@ struct ggml_tensor * ggml_reshape( if (b->grad) { // gradient propagation is not supported - GGML_ASSERT(false); + //GGML_ASSERT(false); } struct ggml_tensor * result = ggml_new_tensor_impl(ctx, a->type, b->n_dims, b->ne, a->data); @@ -12830,7 +12830,7 @@ static void ggml_compute_backward(struct ggml_context * ctx, struct ggml_tensor if (src0->grad) { src0->grad = ggml_add_impl(ctx, src0->grad, - ggml_reshape(ctx, tensor->grad, src1), + ggml_reshape(ctx, tensor->grad, src0->grad), inplace); } if (src1->grad) {