tilelang.jit.adapter.cython.kernel_cache

Classes

CythonKernelCache

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

Module Contents

class tilelang.jit.adapter.cython.kernel_cache.CythonKernelCache

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