tilelang.carver.arch package

Subpackages

• tilelang.carver.arch.driver package
  • Submodules
    • tilelang.carver.arch.driver.cuda_driver module
      • cudaDeviceProp
      • get_cuda_device_properties()
      • get_device_attribute()
      • get_device_name()
      • get_max_dynamic_shared_size_bytes()
      • get_shared_memory_per_block()
  • Module contents

Submodules

• tilelang.carver.arch.arch_base module
  • TileDevice
    • TileDevice.get_avaliable_tensorintrin_shapes()
• tilelang.carver.arch.cdna module
  • CDNA
  • is_cdna_arch()
• tilelang.carver.arch.cpu module
  • CPU
  • is_cpu_arch()
• tilelang.carver.arch.cuda module
  • CUDA
    • CUDA.get_avaliable_tensorintrin_shapes()
  • TensorInstruction
  • check_sm_version()
  • has_mma_support()
  • is_ada_arch()
  • is_ampere_arch()
  • is_cuda_arch()
  • is_hopper_arch()
  • is_tensorcore_supported_precision()
  • is_volta_arch()

Module contents

tilelang.carver.arch.auto_infer_current_arch() → TileDevice

tilelang.carver.arch.get_arch(target: Union[str, Target] = 'cuda') → TileDevice