#0: squash later

tenstorrent · Jan 25, 2024 · e29b71a · e29b71a
1 parent f6fcf22
commit e29b71a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tt_eager/tt_dnn/op_library/conv/kernels/conv_bmm_tilize_col_major_out_blocks.cpp b/tt_eager/tt_dnn/op_library/conv/kernels/conv_bmm_tilize_col_major_out_blocks.cpp
@@ -170,13 +170,13 @@ void MAIN {
     #endif
 
     mm_block_init(mm_in0_cb_id, in1_cb_id, out_cb_id);
+    #ifdef SFPU_OP_INIT_ACTIVATION
+    SFPU_OP_INIT_ACTIVATION
+    #endif
     // in1 num blocks w is the outer loop. Output blocks are computed in col major order.
     for(uint32_t in1_block_w_i = 0; in1_block_w_i < in1_num_blocks_w; ++in1_block_w_i) {
 
         for(uint32_t in0_block_h_i = 0; in0_block_h_i < in0_num_blocks_h; ++in0_block_h_i) {
-            #ifdef SFPU_OP_INIT_ACTIVATION
-            SFPU_OP_INIT_ACTIVATION
-            #endif
 
             #ifdef PRE_TILIZE
             unpack_reconfig_data_format_srca(in1_cb_id, in0_pretilize_cb_id);