tilelang.rocm.intrinsics.utils

Functions

mfma_store_index_map(thread_id, local_id)
mfma_store_index_map_32x32(thread_id, local_id)
get_mma_micro_size(dtype)

Module Contents

tilelang.rocm.intrinsics.utils.mfma_store_index_map(thread_id, local_id)

tilelang.rocm.intrinsics.utils.mfma_store_index_map_32x32(thread_id, local_id)

tilelang.rocm.intrinsics.utils.get_mma_micro_size(dtype)

Parameters:

dtype (Literal['float16', 'int8'])