From c483a7dac5b0bbe400dd857b149dd50fdc03696d Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Wed, 26 Apr 2023 20:34:08 +0200
Subject: [PATCH] bug fix for reshape backward pass

---
 ggml.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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) {