tilelang.ascend.kernel_cache

Ascend-specific full-kernel cache layouts.

Classes

AscendTVMFFIKernelCache	Use source suffixes that describe the exported Ascend artifacts.
AscendCythonKernelCache	Cache the Bisheng device and combined launcher sources as AscendC.
PTOTVMFFIKernelCache	Use source suffixes that describe the exported PTO artifacts.

Module Contents

class tilelang.ascend.kernel_cache.AscendTVMFFIKernelCache

Bases: tilelang.jit.adapter.kernel_cache.TVMFFIKernelCache

Use source suffixes that describe the exported Ascend artifacts.

device_kernel_path = 'device_kernel.asc'

host_kernel_path = 'host_kernel.c'

class tilelang.ascend.kernel_cache.AscendCythonKernelCache

Bases: tilelang.jit.adapter.cython.kernel_cache.CythonKernelCache

Cache the Bisheng device and combined launcher sources as AscendC.

device_kernel_path = 'device_kernel.asc'

host_kernel_path = 'host_kernel.asc'

class tilelang.ascend.kernel_cache.PTOTVMFFIKernelCache

Bases: tilelang.jit.adapter.kernel_cache.TVMFFIKernelCache

Use source suffixes that describe the exported PTO artifacts.

device_kernel_path = 'device_kernel.ptodsl.py'

host_kernel_path = 'host_kernel.c'