tilelang.cuda.target

Functions

check_cuda_availability()	Check if CUDA is available on the system by locating the CUDA path.
normalize_cutedsl_target(target)
target_is_cuda(target)
target_is_volta(target)
target_is_turing(target)
target_is_ampere(target)
target_is_hopper(target)
target_is_sm120(target)
target_has_async_copy(target)
target_has_ldmatrix(target)
target_has_stmatrix(target)
target_has_bulk_copy(target)

Module Contents

tilelang.cuda.target.check_cuda_availability()

Check if CUDA is available on the system by locating the CUDA path. :returns: True if CUDA is available, False otherwise. :rtype: bool

Return type:

bool

tilelang.cuda.target.normalize_cutedsl_target(target)

Parameters:

target (tilelang.backend.target.TargetLike)

Return type:

tvm.target.Target | None

tilelang.cuda.target.target_is_cuda(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_is_volta(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_is_turing(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_is_ampere(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_is_hopper(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_is_sm120(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_has_async_copy(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_has_ldmatrix(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_has_stmatrix(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.cuda.target.target_has_bulk_copy(target)

Parameters:

target (tvm.target.Target)

Return type:

bool