tilelang.jit.adapter.cutedsl.kernel_cache

Classes

CuTeDSLKernelCache

Persist CuTeDSL kernels as Python modules plus launcher artifacts.

Module Contents

class tilelang.jit.adapter.cutedsl.kernel_cache.CuTeDSLKernelCache

Bases: tilelang.cache.kernel_cache.KernelCache

Persist CuTeDSL kernels as Python modules plus launcher artifacts.

kernel_lib_path = 'kernel.py'

device_kernel_path = 'device_kernel.py'

host_kernel_path = 'host_kernel.py'

launcher_lib_path = 'launcher_lib.so'

launcher_cpp_path = 'launcher.cpp'