【合入来源】

任务：

【开源实习】Torch-NPU API一致性对齐：补齐测试用例、API功能对齐、补齐文档（39）#2968
【开源实习】Torch-NPU API一致性对齐：补齐测试用例、API功能对齐、补齐文档（41）#2971

任务分析issue：

【社区任务】: KinetoStepTracker.init_step_count 和 KinetoStepTracker.erase_step_count 的测试覆盖与 NPU 适配检查 #3048

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

本次提交为 torch.autograd.profiler.KinetoStepTracker 的两个方法补充测试用例：

• KinetoStepTracker.init_step_count
• KinetoStepTracker.erase_step_count

【API 功能介绍】

• init_step_count(requester)：为指定请求者初始化步数计数，首次调用时注册并将步数置为 0，重复调用无操作。
• erase_step_count(requester)：移除指定请求者的步数计数记录，全局步数保持历史最大值。

【修改内容】

新增测试文件 test/profiler/test_kineto_step_tracker.py，为原生行为建立回归测试，覆盖以下 7 个核心场景：

【资料变更】

不涉及。

【接口变更】

不涉及。

【功能验证】

测试场景：新增测试用例验证

补丁本地验证 ：

python -c "
import torch
import torch_npu
print(f'torch={torch.__version__}')
print(f'torch_npu={torch_npu.__version__}')
print(f'accelerator={torch.npu.get_device_name() if torch.npu.is_available() else \"cpu\"}')
"
python /workspace/ascend-pytorch/test/profiler/test_kineto_step_tracker.py -v

测试结果：

torch=torch 2.7.1+cpu
torch_npu=torch-npu 2.7.1.post4
accelerator=Ascend910B3
test_erase_all_requesters_keeps_history (__main__.TestKinetoStepTracker.test_erase_all_requesters_keeps_history)
Global step retains historical maximum even after all requesters are erased. ... ok
test_erase_keeps_step_monotonic (__main__.TestKinetoStepTracker.test_erase_keeps_step_monotonic)
Erasing any requester never decreases global step (monotonic contract). ... ok
test_erase_return_value_contract (__main__.TestKinetoStepTracker.test_erase_return_value_contract)
erase_step_count returns bool indicating whether deletion succeeded. ... ok
test_increment_takes_maximum (__main__.TestKinetoStepTracker.test_increment_takes_maximum)
current_step always equals the maximum step among all requesters. ... ok
test_init_does_not_alter_global_step (__main__.TestKinetoStepTracker.test_init_does_not_alter_global_step)
init_step_count only registers requester, never changes global step. ... ok
test_new_requester_inherits_current_step (__main__.TestKinetoStepTracker.test_new_requester_inherits_current_step)
New requester inherits current global step for alignment. ... ok
test_reinit_is_idempotent (__main__.TestKinetoStepTracker.test_reinit_is_idempotent)
Calling init_step_count repeatedly on the same requester has no side effect. ... ok

----------------------------------------------------------------------
Ran 7 tests in 0.003s

OK

【CheckList】

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