Fix init calls for binary sfpu pow and quant (#54)

tenstorrent · Jan 30, 2025 · c5914d8 · c5914d8
1 parent 3969f23
commit c5914d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/common/inc/sfpu/ckernel_sfpu_binary.h b/common/inc/sfpu/ckernel_sfpu_binary.h
@@ -138,7 +138,7 @@ inline void _calculate_sfpu_binary_(const uint dst_offset)
 template <bool APPROXIMATION_MODE /*unused*/, BinaryOp BINOP>
 inline void _sfpu_binary_init_()
 {
-    if constexpr (BINOP == BinaryOp::DIV) {
+    if constexpr (BINOP == BinaryOp::DIV || BINOP == BinaryOp::POW) {
         _init_reciprocal_<APPROXIMATION_MODE>();
     }
 }

diff --git a/common/inc/sfpu/ckernel_sfpu_quant.h b/common/inc/sfpu/ckernel_sfpu_quant.h
@@ -98,6 +98,7 @@ inline void _dequant_int32_(const uint dst_offset)
     }
 }
 
+template <bool APPROXIMATION_MODE /*unused*/>
 inline void _init_quant_zero_point_(const uint zero_point)
 {
     _sfpu_load_imm32_(2,zero_point);
-Original file line number
+Diff line change
@@ Expand Up / @@ -98,6 +98,7 @@ inline void _dequant_int32_(const uint dst_offset) @@
         }
     }
+    template <bool APPROXIMATION_MODE /*unused*/>
     inline void _init_quant_zero_point_(const uint zero_point)
     {
         _sfpu_load_imm32_(2,zero_point);
@@ Expand Down @@