tilelang.rocm.target

Attributes

ROCM_MTRIPLE

Functions

normalize_rocm_arch(arch)
target_get_mcpu(target)
rocm_warp_size_for_arch(arch)
with_rocm_target_attrs(target)
check_hip_availability()	Check if HIP (ROCm) is available on the system by locating the ROCm path.
normalize_rocm_target(target)
target_is_hip(target)
target_is_cdna(target)
target_is_rdna(target)
target_is_gfx950(target)
target_get_warp_size(target)
target_get_rdna_generation(target)

Module Contents

tilelang.rocm.target.ROCM_MTRIPLE = 'amdgcn-amd-amdhsa-hcc'

tilelang.rocm.target.normalize_rocm_arch(arch)

Parameters:

arch (str | None)

Return type:

str | None

tilelang.rocm.target.target_get_mcpu(target)

Parameters:

target (str | tvm.target.Target | None)

Return type:

str | None

tilelang.rocm.target.rocm_warp_size_for_arch(arch)

Parameters:

arch (str | None)

Return type:

int | None

tilelang.rocm.target.with_rocm_target_attrs(target)

Parameters:

target (tvm.target.Target)

Return type:

tvm.target.Target

tilelang.rocm.target.check_hip_availability()

Check if HIP (ROCm) is available on the system by locating the ROCm path. :returns: True if HIP is available, False otherwise. :rtype: bool

Return type:

bool

tilelang.rocm.target.normalize_rocm_target(target)

Parameters:

target (tilelang.backend.target.TargetLike)

Return type:

tvm.target.Target | None

tilelang.rocm.target.target_is_hip(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.rocm.target.target_is_cdna(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.rocm.target.target_is_rdna(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.rocm.target.target_is_gfx950(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.rocm.target.target_get_warp_size(target)

Parameters:

target (tvm.target.Target)

Return type:

int

tilelang.rocm.target.target_get_rdna_generation(target)

Parameters:

target (tvm.target.Target)

Return type:

int