tilelang.jit.adapter.nvrtc.kernel_cache

Classes

NVRTCKernelCache

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

Module Contents

class tilelang.jit.adapter.nvrtc.kernel_cache.NVRTCKernelCache

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.cubin'

kernel_py_path = 'kernel.py'