变更摘要

此 PR 修复了两个与 torch_npu Inductor 后端相关的 bug：一是自动调优（autotune）在调用者已开启 profiler 时会嵌套启动新的 profiler 会话，导致外部会话的 trace 数据被提前终结；二是 filter_masks 在处理动态形状 cat 时，错误地将 size symbol 当作循环轴处理，导致 guard mask 被丢弃、读取越界。修复方案包括新增 profiler 状态探测函数 npu_profiler_session_active() 以在关键路径上跳过嵌套 profiler 启动，以及在 NPUIndexTritonKernel.filter_masks 中过滤掉 size symbol。

主要改动

• 新增 npu_profiler_session_active() 探测函数：在 torch_npu/_inductor/runtime/triton_heuristics.py 中新增该函数，通过访问 ProfPathCreator().is_prof_inited 判断当前进程是否已有活跃的 profiler 会话，并在探测失败时静默返回 False，确保不会中断 kernel 启动路径。
• 批量 benchmark 跳过外部 profiler 会话：在 NPUCachingAutotuner._benchmark_kernel_funcs_batch 中，当 npu_profiler_session_active() 返回 True 时跳过 mspti_batch_benchmark 调用并返回 None，让调用者回退到逐配置的 event timer 方式。
• 带宽 benchmark 回退到 event timer：在 NPUCachingAutotuner._bench_with_launch_args 中，当 profiler 会话活跃时跳过 do_bench_using_profiling_npu，改用 benchmarker.benchmark_gpu 的 event timer 进行计时，避免嵌套 profiler 会话破坏外部 trace。
• filter_masks 过滤 size symbol：在 NPUIndexTritonKernel.filter_masks 中，从 current_subblock_axis 中排除以 s/ps/i 开头的 size symbol，防止其因无法通过 issubset 测试而错误地丢弃动态 cat 切片上的 guard mask。
• 新增完整测试覆盖：新增 test/_inductor/test_autotune_profiler_session.py（5 个用例）和 test/_inductor/test_dynamic_cat_mask.py（5 个用例），分别验证 profiler 嵌套防护的各类回退路径和动态 cat 的 mask 过滤正确性。

openLiBing可接管PR, PR详情链接:
https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=pytorch&number=44117&platform=gitcode&codeHostingPlatformFlag=gitcode

代码审查

✅ 未发现问题

/sync master