【合入来源】

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

关联 Issue：https://gitcode.com/Ascend/pytorch/issues/3596
关联 Issue：https://gitcode.com/Ascend/pytorch/issues/3669

【修改方案】

调用链和现有作用

1. 当设置 TORCH_NPU_SANITIZER 后导入 torch_npu，或显式调用 enable_npu_sanitizer() 时，会进入 NPUSanitizer.enable()。
2. Stream Check 模式在 enable_stream_check() 中创建 NPURecordStreamHandler 和 NPUSanitizerDispatchMode，随后调用 dispatch.__enter__()，将该 Mode 压入当前线程的 TorchDispatchMode 栈。
3. Mode 启用期间，每个被 PyTorch Dispatcher 拦截的普通 ATen 算子都会自动进入 NPUSanitizerDispatchMode.__torch_dispatch__()。该类负责分析算子输入/输出的内存读写、执行真实算子并调用 EventHandler 检查跨 Stream 数据竞争。
4. 旧 enable_autograd(aten_api) 在真实算子执行前调用，原意是在五个白名单名称匹配时从 TLS excluded 集合中移除 AutogradFunctionality，使 Autograd Dispatch Key 重新参与 redispatch。

删减原因

PyTorch 2.13 进入 DispatchMode 的实际名称是 _adaptive_avg_pool2d、native_batch_norm、_log_softmax、nll_loss_forward、_to_copy 等内部 ATen 名称，与旧白名单中的公开 API 名称不相等。

五个真实用例均覆盖到旧 enable_autograd() 调用点，但白名单命中次数和 TLS 状态修改次数均为 0。因此 setter 真分支在当前真实路径中不可达。

代码修改

1. 删除 self.npu_adjust_autograd 白名单。
2. 删除 enable_autograd() 方法。
3. 删除 aten_api 临时变量和 self.enable_autograd(aten_api) 调用。
4. 删除只验证上述旧函数和字段的过期单元测试。
5. 保留 TorchDispatchMode 注册、输入/输出分析、真实算子执行、Stream 获取、数据竞争检查和 record_stream 检查流程。

【资料变更】

不涉及。

【接口变更】

不涉及跨代码仓或客户可见接口变更。本次仅删除 NPU Sanitizer 内部不可达的兼容逻辑。

【功能验证】

当前 master 自检：

• git diff --check 执行通过。
• python -m py_compile torch_npu/npu/_stream_check.py test/npu/test_stream_check.py 执行通过。
• GitCode 远端 Git Hooks 检查通过。

NPU 实机消融验证环境：

• PyTorch：2.13.0a0+gitfad7424
• torch_npu：2.13.0+git19cd51d
• 设备：Ascend 910B2
• 同一 torch_npu master 提交分别构建保留版和删除版 wheel。

验证结果：

• adaptive_avg_pool2d、batch_norm、log_softmax、nll_loss、to 前向和反向均通过。
• 删除前后输出、梯度、shape、dtype、device 逐元素一致。
• BatchNorm running_mean 和 running_var 一致。
• 删除前后有序 Sanitizer Dispatch 序列和各 operator 调用次数一致。
• Sanitizer 启用前、启用后和用例结束后的 TLS 状态一致。
• 五个真实用例白名单命中和 TLS 状态修改次数均为 0。
• torch_npu 官方 Sanitizer 测试：保留版 38/38 通过，删除版 37/37 通过。
• 数量差异来自删除一个只验证旧 enable_autograd() 的过期测试。

【CheckList】

• 代码注释完备，正确记录错误日志
• 代码实现进行了返回值、空指针等校验
• PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
• PR持续集成流水线（CI）执行通过，代码检查无异常