tilelang.engine package

Submodules

• tilelang.engine.callback module
  • register_cuda_postproc()
  • register_cuda_postproc_callback()
  • register_hip_postproc()
  • register_hip_postproc_callback()
• tilelang.engine.lower module
  • canon_target_host()
  • device_codegen()
  • device_codegen_without_compile()
  • extrac_params()
  • get_device_call()
  • get_host_call()
  • has_device_kernel_launch()
  • host_codegen()
  • is_cpu_device_backend()
  • is_device_call()
  • is_device_call_c_device()
  • lower()
• tilelang.engine.param module
  • CompiledArtifact
    • CompiledArtifact.device_mod
    • CompiledArtifact.host_mod
    • CompiledArtifact.kernel_source
    • CompiledArtifact.params
    • CompiledArtifact.rt_mod
  • KernelParam
    • KernelParam.dtype
    • KernelParam.from_buffer()
    • KernelParam.from_var()
    • KernelParam.is_boolean()
    • KernelParam.is_float8()
    • KernelParam.is_scalar()
    • KernelParam.is_unsigned()
    • KernelParam.shape
• tilelang.engine.phase module
  • LowerAndLegalize()
  • OptimizeForTarget()

Module contents