版本兼容性

版本兼容性#

本页按 tpuasm 版本列出支持的组合。所有版本都只支持 Linux x86-64,并需要支持 C++17 的 g++ 编译随包分发的原生桥接。

表中“编解码”指汇编与反汇编,“来源”指 compiler_source_mapping 的编译来源捕获。“—”表示该构建没有这个目标的后端。

未发布#

测试所用 JAX:0.12.0.dev20260926+886d2370c1(commit 886d2370c1c959d210f522e352e3ddc6bcff7d6c),jaxlib 0.11.2。

Python

libtpu

GNU build-id

TPU v4 TC

TPU v4 BCS

TPU v6e TC

TPU v6e TEC

CPython 3.14t

0.0.48

3310a7c8c137cd515c7a2ba1ce2ea38c

编解码、来源

编解码

—

—

CPython 3.14t

0.0.49

97e27df7268da25ab03e455e30dd86b0

编解码、来源

编解码

编解码、来源

编解码

各构建登记的来源 hook 不同:0.0.48 只登记发射、替换、BF16 合并和 store 注释 hook,load 合并、DMA 展开、MXU prep 改写等 hook 只在 0.0.49 登记,所以同一 kernel 在 0.0.48 上有来源的指令更少。

0.2.0#

测试所用 JAX:0.12.0.dev20260926+886d2370c1(commit 886d2370c1c959d210f522e352e3ddc6bcff7d6c),jaxlib 0.11.2。

Python

libtpu

GNU build-id

TPU v4 TC

TPU v4 BCS

TPU v6e TC

TPU v6e TEC

CPython 3.14t

0.0.48

3310a7c8c137cd515c7a2ba1ce2ea38c

编解码、来源

编解码

—

—

CPython 3.14t

0.0.49

97e27df7268da25ab03e455e30dd86b0

编解码、来源

编解码

编解码、来源

编解码

各构建登记的来源 hook 不同:0.0.48 只登记发射、替换、BF16 合并和 store 注释 hook,load 合并、DMA 展开、MXU prep 改写等 hook 只在 0.0.49 登记,所以同一 kernel 在 0.0.48 上有来源的指令更少。

0.1.0#

测试所用 JAX:0.12.0.dev20260926+886d2370c1(commit 886d2370c1c959d210f522e352e3ddc6bcff7d6c),jaxlib 0.11.2。

Python

libtpu

GNU build-id

TPU v4 TC

TPU v4 BCS

TPU v6e TC

CPython 3.14t

0.0.48.dev20260912+nightly

825044f87748f172b7db935904b3f754

编解码、来源

编解码

—

CPython 3.14t

0.0.48

3310a7c8c137cd515c7a2ba1ce2ea38c

编解码、来源

编解码

—

CPython 3.14t

0.0.49

97e27df7268da25ab03e455e30dd86b0

编解码、来源

编解码

编解码、来源

0.0.48 系列只登记发射、替换、BF16 合并和 store 注释 hook,load 合并 hook 只在 0.0.49 登记。