File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ def decompose_col_major(index_1d: int, basis: List[int]) -> List[int]:
2121
2222def _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
108108def _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 """
You can’t perform that action at this time.
0 commit comments