From 1f2b76de01ce3da98417518d4fad6f5d1fa89f6f Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Sun, 14 May 2023 20:55:24 +0200
Subject: [PATCH] fix bug in ggml_compute_forward_soft_max_back_f32 on DEBUG
 build

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

diff --git a/ggml.c b/ggml.c
index 2286c615d..06e3feea0 100644
--- a/ggml.c
+++ b/ggml.c
@@ -10565,8 +10565,8 @@ static void ggml_compute_forward_soft_max_back_f32(
 #ifndef NDEBUG
         for (int i = 0; i < nc; ++i) {
             //printf("p[%d] = %f\n", i, p[i]);
-            assert(!isnan(s0[i]));
-            assert(!isnan(s1[i]));
+            assert(!isnan(dy[i]));
+            assert(!isnan(y[i]));
         }
 #endif
         // Jii = yi - yi*yi