版本兼容性#
本页按 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 |
|
|
编解码、来源 |
编解码 |
— |
— |
CPython 3.14t |
|
|
编解码、来源 |
编解码 |
编解码、来源 |
编解码 |
各构建登记的来源 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 |
|
|
编解码、来源 |
编解码 |
— |
— |
CPython 3.14t |
|
|
编解码、来源 |
编解码 |
编解码、来源 |
编解码 |
各构建登记的来源 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 |
|
|
编解码、来源 |
编解码 |
— |
CPython 3.14t |
|
|
编解码、来源 |
编解码 |
— |
CPython 3.14t |
|
|
编解码、来源 |
编解码 |
编解码、来源 |
0.0.48 系列只登记发射、替换、BF16 合并和 store 注释 hook,load 合并 hook 只在 0.0.49 登记。