enable matmul kernel workaround

tenstorrent · Feb 24, 2025 · 72712db · 72712db
1 parent 6d73e0d
commit 72712db
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/...tnn/operations/matmul/device/kernels/compute/bmm_large_block_zm_fused_bias_activation.cpp b/...tnn/operations/matmul/device/kernels/compute/bmm_large_block_zm_fused_bias_activation.cpp
@@ -199,15 +199,13 @@ void MAIN {
                                 // in0_block_w is passed as innder dim (kt) to matmul_block, interally used to stride
                                 // in0
 
-                                // #ifdef ARCH_BLACKHOLE
-                                //                                 // FIXME: This is a temporary workaround to avoid
-                                //                                 hangs on blackhole.
-                                //                                 //
-                                //                                 https://github.com/tenstorrent/tt-metal/issues/16439
-                                //                                 for (uint32_t i = 0; i < 10; i++) {
-                                //                                     asm volatile("nop");
-                                //                                 }
-                                // #endif
+#ifdef ARCH_BLACKHOLE
+                                // FIXME: This is a temporary workaround to avoid hangs on blackhole.
+                                // https://github.com/tenstorrent/tt-metal/issues/16439
+                                for (uint32_t i = 0; i < 100; i++) {
+                                    asm volatile("nop");
+                                }
+#endif
                                 matmul_block(
                                     in0_cb_id,
                                     in1_cb_id,