#6722: Add condition for asin op

tenstorrent · Apr 3, 2024 · 4df7d31 · 4df7d31
1 parent 13edc65
commit 4df7d31
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
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
@@ -307,7 +307,10 @@ inline void calculate_asin()
     for (int d = 0; d < ITERATIONS; d++)
     {
         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++;
     }