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()

Clears the entire kernel cache (using KernelCache class).

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 (Union[str, tvm.target.Target])

  • target_host (Union[str, tvm.target.Target])

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

  • verbose (Optional[bool])

  • pass_configs (Optional[dict])

  • compile_flags (Optional[List[str]])

Return type:

tilelang.jit.JITKernel

tilelang.cache.clear_cache()¶

Clears the entire kernel cache (using KernelCache class).