tilelang.carver.arch.rdna¶

Classes¶

Functions¶

is_rdna_arch(arch)

Module Contents¶

tilelang.carver.arch.rdna.is_rdna_arch(arch)¶

Parameters:: arch (tilelang.carver.arch.arch_base.TileDevice)
Return type:: bool

class tilelang.carver.arch.rdna.RDNA(target)¶

Bases: tilelang.carver.arch.arch_base.TileDevice

Parameters:: target (tvm.target.Target | str)

target¶

rdna_generation¶

device: tvm.runtime.Device¶

platform: str = 'RDNA'¶

smem_cap¶

compute_max_core¶

warp_size = 32¶

compute_capability¶

reg_cap: int = 32768¶

max_smem_usage: int¶

sm_partition: int = 4¶

l2_cache_size_bytes: int¶

transaction_size: list[int] = [32, 128]¶

bandwidth: list[int] = [750, 12080]¶

available_tensor_instructions: list[tilelang.carver.arch.cuda.TensorInstruction] | None = None¶

get_avaliable_tensorintrin_shapes()¶

__repr__()¶