From 93bdbc656dd644819b6ba58710346fd33188fa2c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phane=20du=20Hamel?= <stephduh@live.fr>
Date: Sun, 8 Dec 2024 17:39:57 +0100
Subject: [PATCH] Vulkan: fix NaN in tanh.comp

---
 ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp b/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp
index 74630dc7f..83e2767ba 100644
--- a/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp
+++ b/ggml/src/ggml-vulkan/vulkan-shaders/tanh.comp
@@ -16,6 +16,6 @@ void main() {
     if (i >= p.KX) {
         return;
     }
-
-    data_d[i] = D_TYPE(tanh(data_a[i]));
+    float th = tanh(data_a[i]);
+    data_d[i] = D_TYPE(th==th ? th : sign(data_a[i]));
 }