tilelang.cache package

Submodules

• tilelang.cache.kernel_cache module
  • KernelCache
    • KernelCache.cached()
    • KernelCache.clear_cache()

Module contents

The cache utils with class and database persistence - Init file

tilelang.cache.cached(func: Optional[PrimFunc] = None, out_idx: Optional[List[int]] = None, *args, target: Union[str, Target] = 'auto', target_host: Optional[Union[str, Target]] = None, execution_backend: Optional[Literal['dlpack', 'ctypes', 'cython']] = 'cython', verbose: Optional[bool] = False, pass_configs: Optional[dict] = None) → JITKernel

Caches and reuses compiled kerne(ls (using KernelCache class).

tilelang.cache.clear_cache()

Clears the entire kernel cache (using KernelCache class).