PTX(并行线程执行)

定义

PTX(Parallel Thread Execution,并行线程执行)是 NVIDIA 为 CUDA 定义的虚拟指令集架构(ISA)与低级中间表示。CUDA C++ 等高级语言先被编译为 PTX,再由驱动或工具链针对具体 GPU 代际编译成实际机器码 SASS。

作用

• 向前兼容:应用只需携带 PTX,新一代 GPU 的驱动可即时(JIT)将其编译为本代机器码,老程序无需重新发布即可在新硬件运行。
• 编译目标:LLVM 的 NVPTX 后端使第三方语言(Julia、Triton、Mojo 等)都能生成 PTX,从而接入 CUDA 生态。
• 底层优化:开发者可在 CUDA 代码中内联 PTX 汇编,精确控制指令选择。

与 SASS 的关系

PTX 是稳定的虚拟层,SASS 是随架构变化的真实指令集;性能分析工具反汇编看到的最终执行代码是 SASS。

意义

PTX 这一稳定中间层是 CUDA 生态护城河的重要组成:它隔离了软件与快速迭代的硬件,美国出口管制文件甚至曾以能否执行特定精度运算的指令能力作为管控参考,足见其行业地位。