[Bugfix] B share to local warps num

bba13746 · wangziyang · 64179eaf · bba13746
Commit bba13746 authored Apr 27, 2026 by wangziyang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

tilelang/intrinsics/mmac_layout.py tilelang/intrinsics/mmac_layout.py +1 -4

No files found.
--- a/tilelang/intrinsics/mmac_layout.py
+++ b/tilelang/intrinsics/mmac_layout.py
@@ -49,10 +49,7 @@ def shared_to_local_layout_B_padding(row, col, idx, warp_cols, block_col_warps,
    
    # return new_row, new_col
    
-    if block_col_warps > 1:
-        inter_idx_padding = 2
-    else:
-        inter_idx_padding = 1
+    inter_idx_padding = block_col_warps
    
    paddings = 32
    new_col += idx * inter_idx_padding * paddings