自动微分2025 年 12 月 18 日约 8 分钟阅读

TPX：计算与梯度如何解耦

微分引擎不必侵入计算核心。组合、监听与调度就可以建立完整动态图。

TensorPlay 工程团队

外挂式封装

tpx::Tensor 组合 p10::Tensor，而不向 P10 张量加入梯度字段。纯计算可以直接使用 P10，保持零额外微分开销。

requires_grad 为 true 时，TPX 记录输入、输出、GradFn 与算子参数并建立 DAG；关闭时，前向路径与纯 P10 一致。

backward() 先拓扑排序，再调度每个节点的 GradFn。TPX 决定梯度规则，具体张量运算仍由 P10 执行。

自定义 GradFn 不需要修改 P10。新硬件只需适配计算层，微分模块可以继续复用。