#3908: Fix sfpu init for exp2,expm1,rsqrt,atan

tenstorrent · Dec 6, 2023 · 36fe1e2 · 36fe1e2
1 parent 6d8c276
commit 36fe1e2
Showing 1 changed file with 13 additions and 19 deletions.
diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_init.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_init.h
@@ -34,30 +34,24 @@ inline void llk_math_eltwise_unary_sfpu_init(
     _llk_math_eltwise_unary_sfpu_init_();
 
     switch (sfpu_op) {
-        case SfpuType::reciprocal:
-            sfpu::_init_reciprocal_<APPROXIMATE>();
-            break;
-        case SfpuType::exponential:
-            sfpu::_init_exponential_<APPROXIMATE>();
-            break;
-        case SfpuType::log:
-            sfpu::_init_log_<APPROXIMATE>();
-            break;
-        case SfpuType::sqrt:
-            sfpu::_init_sqrt_<APPROXIMATE>();
-            break;
         case SfpuType::tanh:
         case SfpuType::tanh_derivative:
-            sfpu::_init_tanh_<APPROXIMATE>();
-            break;
+             sfpu::_init_tanh_<APPROXIMATE>();
+             break;
         case SfpuType::sigmoid:
-            sfpu::_init_sigmoid_<APPROXIMATE>();
+             sfpu::_init_sigmoid_<APPROXIMATE>();
+             break;
+        case SfpuType::expm1:
+        case SfpuType::exp2:
+            sfpu::_init_exponential_<APPROXIMATE>();
             break;
-        case SfpuType::gelu_derivative:
-            sfpu::_init_gelu_derivative_<APPROXIMATE>();
+        case SfpuType::rsqrt:
+        case SfpuType::atan:
+            sfpu::_init_reciprocal_<APPROXIMATE>();
             break;
-        case SfpuType::gelu:
-            sfpu::_init_gelu_<APPROXIMATE>();
+        case SfpuType::log_with_base:
+        case SfpuType::log:
+            sfpu::_init_log_<APPROXIMATE>();
             break;
         case SfpuType::dropout:
             sfpu::_init_dropout_(param2);