viable/strict/1790184407: Support tensor torch._assert in export (#186306)
PyTorch 发布记录显示，torch._assert 此前只对静态为真的 Python 常量和符号表达式做 Dynamo 特殊处理；张量值条件会落到 eager Python 实现，其 if not condition 会强制代理张量走 Python 真值判断，导出时产生 Eq(u0, 1) 这类数据相关 guard，而非在导出程序中保留断言。该变更在 strict export 与非 strict export 中把张量值 torch._assert(condition, message) 降级为 torch._assert_async(condition, message)，静态为真的断言仍照旧省略，修复 #134533。
发展脉络
- 首次出现viable/strict/1790184407: Support tensor torch._assert in export (#186306)PyTorch Core
- 当前判断这是编译器与导出路径的语义一致性修补，而非能力发布。它反映 PyTorch 在 strict 与非 strict 两条导出路径上统一断言处理，减少数据相关 guard 对导出可移植性的干扰；对依赖 torch.export 的下游工具链而言，属于稳定性维护类变更。Agent Pulse · 分析
PyTorch 的 viable/strict/1790184407 发布条目记录了一项导出行为修复：张量值 torch._assert 在导出时不再被主机端求值为数据相关 guard，而是降级为 torch._assert_async，与已用于编译 Python 断言的运行时断言行为一致。测试计划覆盖 test_export.py 的 test_torch_assert_tensor_condition、test_dynamo 的 test_export_persist_assert 与 test_utils 的 TestAssert，并跑了 lintrunner。基准在 main 的 30326da 与本分支上以 3 次预热、12 次测量对比导出延迟，strict_static_torch_assert 中位数由 0.030381s 变为 0.031370s。
从证据看，修复点在于避免导出阶段对数据相关张量谓词做主机端判定：把断言下沉为异步运行时断言，可让导出程序保留断言语义而非生成 Eq(u0, 1) 式 guard。可验证的下一信号是导出产物中是否不再出现该 guard，以及 torch._assert_async 在导出图中的表示是否稳定。
这是编译器与导出路径的语义一致性修补，而非能力发布。它反映 PyTorch 在 strict 与非 strict 两条导出路径上统一断言处理，减少数据相关 guard 对导出可移植性的干扰；对依赖 torch.export 的下游工具链而言，属于稳定性维护类变更。
对使用 torch.export 做部署的团队，减少数据相关 guard 有助于降低导出图对具体输入形状或取值的依赖，从而减少重编译与运行时分歧风险。证据未给出生产环境收益数据，价值判断需以实际导出产物与延迟基准复核。
若该降级策略在后续版本保持，导出图中数据相关断言将更多以异步运行时断言形式存在。可观察的下一信号是相关测试是否长期保留，以及是否有后续条目调整 torch._assert_async 的导出语义。