tilelang.cache¶

The cache utils with class and database persistence - Init file

Submodules¶

Functions¶

cached([func, out_idx, target, target_host, ...])

Caches and reuses compiled kernels (using KernelCache class).

clear_cache()

Disabled helper that previously removed the entire kernel cache.

Package Contents¶

tilelang.cache.cached(func=None, out_idx=None, *args, target='auto', target_host=None, execution_backend='cython', verbose=False, pass_configs=None, compile_flags=None)¶

Caches and reuses compiled kernels (using KernelCache class).

Parameters:

  • func (tvm.tir.PrimFunc)

  • out_idx (list[int])

  • target (str | tvm.target.Target)

  • target_host (str | tvm.target.Target)

  • execution_backend (Literal['dlpack', 'ctypes', 'cython', 'nvrtc'] | None)

  • verbose (bool | None)

  • pass_configs (dict | None)

  • compile_flags (list[str] | str | None)

Return type:

tilelang.jit.JITKernel

tilelang.cache.clear_cache()¶

Disabled helper that previously removed the entire kernel cache.

Raises:

RuntimeError – Always raised to warn users to clear the cache manually.