tilelang.carver.arch.cuda module

class tilelang.carver.arch.cuda.CUDA(target: Union[Target, str])

Bases: TileDevice

get_avaliable_tensorintrin_shapes()

class tilelang.carver.arch.cuda.TensorInstruction(name: str, shape: List[int])

Bases: object

tilelang.carver.arch.cuda.check_sm_version(arch: str) → int

tilelang.carver.arch.cuda.has_mma_support(arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_ada_arch(arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_ampere_arch(arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_cuda_arch(arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_hopper_arch(arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_tensorcore_supported_precision(in_dtype: str, accum_dtype: str, arch: TileDevice) → bool

tilelang.carver.arch.cuda.is_volta_arch(arch: TileDevice) → bool