Revert "#13415: Update left shift (#15538)"

This reverts commit 7b74c5b.
tenstorrent · Feb 5, 2025 · d9e8a84 · d9e8a84
1 parent 886377c
commit d9e8a84
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_sfpu/ckernel_sfpu_left_shift.h b/tt_metal/hw/ckernels/blackhole/metal/llk_api/llk_sfpu/ckernel_sfpu_left_shift.h
@@ -17,9 +17,13 @@ template <bool APPROXIMATION_MODE, int ITERATIONS = 8>
 inline void calculate_left_shift(const uint shift_amt) {
 #pragma GCC unroll 0
     for (int d = 0; d < ITERATIONS; d++) {
-        TTI_SFPLOAD(0,12,ADDR_MOD_7,0);
-        TT_SFPSHFT(shift_amt,0,0,1);
-        TTI_SFPSTORE(0,12,ADDR_MOD_7,0);
+        vInt val = dst_reg[0];
+        vInt v = val;
+
+        val = val << shift_amt;
+        val = setsgn(val, v);
+        dst_reg[0] = val;
+
         dst_reg++;
     }
 }

diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_left_shift.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_left_shift.h
@@ -17,9 +17,13 @@ template <bool APPROXIMATION_MODE, int ITERATIONS = 8>
 inline void calculate_left_shift(const uint shift_amt) {
 #pragma GCC unroll 0
     for (int d = 0; d < ITERATIONS; d++) {
-        TTI_SFPLOAD(0,4,3,0);
-        TT_SFPSHFT(shift_amt,0,0,1);
-        TTI_SFPSTORE(0,4,3,0);
+        vInt val = dst_reg[0];
+        vInt v = val;
+
+        val = val << shift_amt;
+        val = setsgn(val, v);
+        dst_reg[0] = val;
+
         dst_reg++;
     }
 }