tilelang.carver.arch.cdna

Classes

CDNA

Functions

is_cdna_arch(arch)

Module Contents

tilelang.carver.arch.cdna.is_cdna_arch(arch)

Parameters:

arch (tilelang.carver.arch.arch_base.TileDevice)

Return type:

bool

class tilelang.carver.arch.cdna.CDNA(target)

Bases: tilelang.carver.arch.arch_base.TileDevice

Parameters:

target (tvm.target.Target | str)

target

device: tvm.runtime.Device

platform: str = 'CDNA'

smem_cap

compute_max_core

warp_size

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] = [1300, 14000]