tilelang.cache

The cache utils with class and database persistence - Init file

子模組

函式

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=None, target_host=None, execution_backend=None, verbose=None, pass_configs=None, compile_flags=None)

Caches and reuses compiled kernels (using KernelCache class).

參數:

  • func (tvm.tir.PrimFunc)

  • out_idx (list[int])

  • target (str | tvm.target.Target | None)

  • target_host (str | tvm.target.Target | None)

  • execution_backend (Literal['auto', 'tvm_ffi', 'cython', 'nvrtc', 'torch', 'cutedsl'] | None)

  • verbose (bool | None)

  • pass_configs (dict | None)

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

回傳型別:

tilelang.jit.JITKernel

tilelang.cache.clear_cache()

Disabled helper that previously removed the entire kernel cache.

引發:

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