tilelang.layout.gemm_sp

Wrapping Layouts.

Functions

decompose_col_major(index_1d, basis)
make_metadata_layout(buffer[, mma_dtype, arch, backend])

Module Contents

tilelang.layout.gemm_sp.decompose_col_major(index_1d, basis)

Parameters:

• index_1d (int)

• basis (List[int])

Return type:

List[int]

tilelang.layout.gemm_sp.make_metadata_layout(buffer, mma_dtype='float16', arch='sm90', backend='cutlass', **extra_args)

Parameters:

• buffer (tvm.tir.Buffer)

• mma_dtype (str)

• arch (str)

• backend (str)