Skip to content

Commit 122bff0

Browse files
committed
[Cleanup] Remove unnecessary blank lines in metadata layout functions in gemm_sp.py
1 parent 333e4d4 commit 122bff0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tilelang/layout/gemm_sp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def decompose_col_major(index_1d: int, basis: List[int]) -> List[int]:
2121

2222
def _make_metadata_layout_sm90_cutlass(buffer: tvm.tir.Buffer, mma_dtype: str, block_k: int):
2323
"""Make a layout of metadata that is compatible with cutlass sm90 compression kernel. Note that layout atom is the same for smem and gmem.
24-
24+
2525
Args:
2626
buffer: metadata buffer shape, for sm90 it should be a 8-bit type
2727
mma_dtype: dtype of mma operand A, different dtypes result in different layout atom
@@ -107,7 +107,7 @@ def transform(i: int, k: int) -> int:
107107

108108
def _make_metadata_layout_sm8x_cutlass(buffer: tvm.tir.Buffer, mma_dtype: str):
109109
"""Make a layout of metadata that is compatible with cutlass sm8x compression kernel. Note that layout atom is the same for smem and gmem.
110-
110+
111111
Args:
112112
buffer: metadata buffer shape, for sm80 it should be a 16bit type
113113
"""

0 commit comments

Comments
 (0)