tilelang.jit.adapter.torch.kernel_cache¶
Classes¶
Caches compiled kernels using a class and database persistence to avoid redundant compilation. |
Module Contents¶
- class tilelang.jit.adapter.torch.kernel_cache.TorchKernelCache¶
Bases:
tilelang.cache.kernel_cache.KernelCacheCaches 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