Skip to content

Commit a6bab65

Browse files
committed
bugfix
1 parent f7fe22d commit a6bab65

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tilelang/intrinsics/wgmma_macro_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ def _warp_mma(A_buf, B_ptr, C_buf):
401401
for i in T.unroll(num_inst_m):
402402
for ki in T.unroll(0, (k_dim // micro_size_k)):
403403
warp_j = warp_n * num_inst_n + j
404-
scale_out = T.select(ki != 0, 1, T.select(clear_accum, 0, 1))
404+
scale_out = T.Select(ki != 0, 1, T.Select(clear_accum, 0, 1))
405405

406406
A_offset = ki * warp_rows * local_size_a + i * local_size_a
407407
B_offset = (

0 commit comments

Comments
 (0)