#6721: Fix log nan issue

tenstorrent · Sep 23, 2024 · 792f99d · 792f99d
1 parent 96fd0df
commit 792f99d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/tests/tt_eager/python_api_testing/unit_testing/misc/test_pow_fractional.py b/tests/tt_eager/python_api_testing/unit_testing/misc/test_pow_fractional.py
@@ -6,10 +6,12 @@
 import torch
 
 import ttnn
-from models.utility_functions import comp_pcc
+from models.utility_functions import comp_pcc, skip_for_grayskull, skip_for_wormhole_b0
 from loguru import logger
 
 
+@skip_for_grayskull("https://github.com/tenstorrent/tt-metal/issues/12776")
+@skip_for_wormhole_b0("https://github.com/tenstorrent/tt-metal/issues/12776")
 def test_pow_fractional_composite(device):
     torch.manual_seed(577215)
     N = 1

diff --git a/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_sfpu/ckernel_sfpu_log.h b/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_sfpu/ckernel_sfpu_log.h
@@ -61,6 +61,11 @@ sfpi_inline void calculate_log_body(const uint log_base_scale_factor) {
     }
     v_endif;
 
+    v_if (in < 0.0F) {
+        result = std::numeric_limits<float>::quiet_NaN();
+    }
+    v_endif;
+
     dst_reg[0] = result;
 }
 

diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_log.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_log.h
@@ -65,6 +65,11 @@ sfpi_inline void calculate_log_body(const uint log_base_scale_factor)
     }
     v_endif;
 
+    v_if (in < 0.0F) {
+        result = std::numeric_limits<float>::quiet_NaN();
+    }
+    v_endif;
+
     dst_reg[0] = result;
 }