Skip to content

GitLab

Commit bba13746 authored by wangziyang's avatar wangziyang
Browse files

[Bugfix] B share to local warps num

parent 64179eaf
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
tilelang/intrinsics/mmac_layout.py
View file @ bba13746
...@@ -49,10 +49,7 @@ def shared_to_local_layout_B_padding(row, col, idx, warp_cols, block_col_warps, ...@@ -49,10 +49,7 @@ def shared_to_local_layout_B_padding(row, col, idx, warp_cols, block_col_warps,
# return new_row, new_col # return new_row, new_col
if block_col_warps > 1: inter_idx_padding = block_col_warps
inter_idx_padding = 2
else:
inter_idx_padding = 1
paddings = 32 paddings = 32
new_col += idx * inter_idx_padding * paddings new_col += idx * inter_idx_padding * paddings
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment