tilelang.cache.cuda_binary_cache¶
Cross-host cache for compiled CUDA device binaries.
Classes¶
Cache cubin/fatbin bytes independently from host executable artifacts. |
Module Contents¶
- class tilelang.cache.cuda_binary_cache.CUDABinaryCache¶
Cache cubin/fatbin bytes independently from host executable artifacts.
- cache_root_dir = 'cuda-binaries'¶
- classmethod make_key(*, code, target_kind, target_arch, target_code, compile_format)¶
- Parameters:
code (str)
target_kind (str)
target_arch (str)
target_code (list[str])
compile_format (str)
- Return type:
str
- classmethod get_path(key, compile_format)¶
- Parameters:
key (str)
compile_format (str)
- Return type:
str
- classmethod load(key, compile_format)¶
- Parameters:
key (str)
compile_format (str)
- Return type:
bytes | None
- classmethod save(key, compile_format, data)¶
- Parameters:
key (str)
compile_format (str)
data (bytes)
- Return type:
None