#6722: Add condition for asin op

tenstorrent · Apr 10, 2024 · 3f520c3 · 3f520c3
1 parent c419751
commit 3f520c3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_sfpu/ckernel_sfpu_trigonometry.h b/tt_metal/hw/ckernels/grayskull/metal/llk_api/llk_sfpu/ckernel_sfpu_trigonometry.h
@@ -300,6 +300,10 @@ inline void calculate_asin()
     {
         vFloat v = dst_reg[0];
         v = sfpu_asine_maclaurin_series<APPROXIMATION_MODE>(v);
+        v_if(dst_reg[0] < -1.0f || dst_reg[0] > 1.0f )
+        {
+           v = std::numeric_limits<float>::quiet_NaN();
+        }v_endif;
         dst_reg[0] = v;
         dst_reg++;
     }

diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_trigonometry.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_trigonometry.h
@@ -308,6 +308,10 @@ inline void calculate_asin()
     {
         vFloat v = dst_reg[0];
         v = sfpu_asine_maclaurin_series<APPROXIMATION_MODE>(v);
+        v_if(dst_reg[0] < -1.0f || dst_reg[0] > 1.0f )
+        {
+           v = std::numeric_limits<float>::quiet_NaN();
+        }v_endif;
         dst_reg[0] = v;
         dst_reg++;
     }