tilelang.cache.ascend_binary_cache¶

Cross-host caches for compiled Ascend and PTO device binaries.

Classes¶

AscendBinaryCache

Cache executable CCE ELF bytes independently from host artifacts.

PTOBinaryCache

Cache executable PTODSL-compiled CCE ELF bytes for the PTO target.

Module Contents¶

class tilelang.cache.ascend_binary_cache.AscendBinaryCache¶

Bases: tilelang.cache.binary_cache.BinaryCache

Cache executable CCE ELF bytes independently from host artifacts.

cache_root_dir = 'ascend-binaries'¶
cache_format = 'tilelang.ascend-binary-cache.v1'¶
binary_kind = 'Ascend'¶
binary_format = 'aibin'¶
classmethod make_key(*, code, target_kind, target_arch, compile_format, options=None, linker_options=None, compiler='bisheng')¶
Parameters:

  • code (str)

  • target_kind (str)

  • target_arch (str)

  • compile_format (str)

  • options (list[str] | None)

  • linker_options (list[str] | None)

  • compiler (str)

Return type:

str

class tilelang.cache.ascend_binary_cache.PTOBinaryCache¶

Bases: tilelang.cache.binary_cache.BinaryCache

Cache executable PTODSL-compiled CCE ELF bytes for the PTO target.

cache_root_dir = 'pto-binaries'¶
cache_format = 'tilelang.pto-binary-cache.v1'¶
binary_kind = 'PTO'¶
binary_format = 'aibin'¶
classmethod make_key(*, code, target_kind, target_arch, compile_format, options=None, linker_options=None, toolchain=None)¶
Parameters:

  • code (str)

  • target_kind (str)

  • target_arch (str)

  • compile_format (str)

  • options (list[str] | None)

  • linker_options (list[str] | None)

  • toolchain (dict[str, str] | None)

Return type:

str