tilelang.jit.adapter.cutedsl.kernel_cache

Classes

CuTeDSLKernelCache

Caches compiled kernels using a class and database persistence to avoid redundant compilation.

Module Contents

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

Bases: tilelang.cache.kernel_cache.KernelCache

Caches compiled kernels using a class and database persistence to avoid redundant compilation. Cache files:

kernel.cu: The compiled kernel source code wrapped_kernel.cu: The compiled wrapped kernel source code kernel_lib.so: The compiled kernel library params.pkl: The compiled kernel parameters

kernel_lib_path = 'kernel.py'

device_kernel_path = 'kernel.py'

host_kernel_path = 'kernel.py'

launcher_lib_path = 'launcher_lib.so'

launcher_cpp_path = 'launcher.cpp'