From 8aa711ad986adf70e90963463ab8984661167f78 Mon Sep 17 00:00:00 2001
From: Molly Sophia <mollysophia379@gmail.com>
Date: Mon, 12 Aug 2024 09:29:47 +0800
Subject: [PATCH] ggml: Add backward computation for unary op ``exp``

Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
---
 ggml/src/ggml.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
index 06d8a8654..a32cfcb09 100644
--- a/ggml/src/ggml.c
+++ b/ggml/src/ggml.c
@@ -18843,6 +18843,15 @@ static void ggml_compute_backward(struct ggml_context * ctx, struct ggml_tensor
                                         zero_table);
                             }
                         } break;
+                    case GGML_UNARY_OP_EXP:
+                        {
+                            if (src0->grad) {
+                                src0->grad = ggml_add_or_set(ctx,
+                                        src0->grad,
+                                        ggml_mul(ctx, tensor, tensor->grad),
+                                        zero_table);
+                            }
+                        } break;
                     default:
                         GGML_ABORT("fatal error");
                 }