tilelang.intrinsics.utils module#

tilelang.intrinsics.utils.get_ldmatrix_offset(matrix: Literal['A', 'B'], row_idx, col_idx, stride, dtype: Literal['float16', 'int8'] = 'float16', transposed: bool = False)#

tilelang.intrinsics.utils.get_mma_micro_size(dtype: Literal['float16', 'int8'])#

tilelang.intrinsics.utils.index_to_coordinates(index, shape)#

General Implementation of:: vjj = index % (micro_size_k // num_elems_per_byte) coordinates[-1] = index % shape[-1]; vii = index // (micro_size_k // num_elems_per_byte) % micro_size_y index = index // shape[-1]; coordinates[-2] = index % shape[-2]; vj = index // (micro_size_k // num_elems_per_byte * micro_size_y) % block_K // (micro_size_k // num_elems_per_byte) index = index // shape[-2]; coordinates[-3] = index % shape[-3]; vi = index // (micro_size_k // num_elems_per_byte * micro_size_y * (block_K // (micro_size_k // num_elems_per_byte))) % block_N // micro_size_y index = index // shape[-3]; coordinates[-4] = index % shape[-4];