tilelang.cache.cuda_binary_cache¶

Cross-host cache for compiled CUDA device binaries.

Classes¶

CUDABinaryCache

Cache cubin/fatbin bytes independently from host executable artifacts.

Module Contents¶

class tilelang.cache.cuda_binary_cache.CUDABinaryCache¶

Bases: tilelang.cache.binary_cache.BinaryCache

Cache cubin/fatbin bytes independently from host executable artifacts.

cache_root_dir = 'cuda-binaries'¶
cache_format = 'tilelang.cuda-binary-cache.v1'¶
binary_kind = 'CUDA'¶
classmethod make_key(*, code, target_kind, target_arch, target_code, compile_format, options=None, compiler='nvcc')¶
Parameters:
  • code (str)

  • target_kind (str)

  • target_arch (str)

  • target_code (list[str])

  • compile_format (str)

  • options (list[str] | None)

  • compiler (str)

Return type:

str