已合并
[sync] PR-39802: [fix]profiler fix config cache when analyse multi card #39874
[sync] PR-39802: [fix]profiler fix config cache when analyse multi card #39874
已合并
ascend-robot创建于 7月2日
ascend-robot
ascend-robot成员
7月2日

1. Origin pull request:

https://gitcode.com/Ascend/pytorch/merge_requests/39802

https://gitcode.com/Ascend/pytorch/issues/2573

Sha Datetime Message
74141fe8 2026-07-01 19:36:24 +0800 CST [fix]profiler fix config cache when analyse multi card
likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 ascend-robot 的贡献)
ascend-robotascend-robot成员
7月2日 创建了 pull request,commit 0d30c915
ascend-robotascend-robot成员
7月2日 关联了issue:[Bug]: 离线解析设置的max_process_number的值小于总ascend_pt数量时,解析出来db文件的rank与profiler_info.json的rank不一致
ascend-robotascend-robot成员
7月2日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
7月2日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
test renyujin, 陈豪 (2/2) 陈豪 (1/1)
torch_npu/profiler 陈豪, renyujin (2/2) 陈豪 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

ascend-ds-bot, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
ascend-robotascend-robot成员
7月2日 添加了label:ci-pipeline-running
ascend-robot
ascend-robot成员
7月2日 评论:

ascend docs pipeline is running...

likedislike
ascend-robotascend-robot成员
7月2日 添加了label:docs-ci-pipeline-running
ascend-robot
ascend-robot成员
7月2日 评论:

✅ 跳过 docs ci 检查,没有需要检查的文档文件

likedislike
ascend-robotascend-robot成员
7月2日 删除了label:docs-ci-pipeline-running
ascend-robotascend-robot成员
7月2日 添加了label:docs-ci-pipeline-success
ascend-robotascend-robot成员
7月2日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
7月2日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
7月2日 评论:
流水线 PR-pipeline_pytorch#40695 [ commitID:312390ac ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_LibTorch_ARM >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
patch_test 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
CodeCheck_lintrunner >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
UT_ARM_A2_Part_03 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_DIST_ARM_Part_01 🛑 >>>
UT_DIST_ARM_Part_02 🛑 >>>
UT_DIST_ARM_Part_03 🛑 >>>
UT_DIST_ARM_Part_04 🛑 >>>
UT_ARM_A2_Select_Part_01 >>>
UT_ARM_A2_Select_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
此流水线已支持下列评论快捷指令,仅PR创建者和白名单成员[wujinyuan1, huangjingwei, liangsongwei, yashi999, culechan, Dring, wuyouqi1, L1919_snow, qq_52711437, WhiteNight12, nomiz, xiu_21, ffmh, wanglijun55, hss-shuai, husichao, smallsilly, lanshaozuishuai, jimmyisme1, lzy0920232, alpha-junh, Sunshine_Youngster, wei_zhuoyi, zhangyihuiben, zyw-hw, zzzkeke, rmch, yangch0324, LucciC, AACAES, renyujin, wjlflyer, senzhen-town, pengjingyou, qsc97, limuan, yule100, xiaoqi-zhou, kuhn7, chenxingying, hanye02, zichun_ye, anyrenwei, kkjocker, wangzili121, Lu_G, yvjc, puddingfjz, HandsoemLemon, bigprestigee1, huawuyi, zhenyu10, dairenjie, du-jin-hang, zou-jieyu, adelaideliu, TrHan, wanlinan, Windwindzzz, pengqihw, kisnwang, yuheng_wang, honghao_wang, jizewei, zhangguoguang, sunyu-xuan, chenrayray, hbhu_bin, liujunzhu, c_34, LiNuoh, maoyuanpeng1, zzhongmin, zhaoyu65, bellatan, jiabaolin, zhuofanshen, wencaiwen, lu_zhuge, caoshuyang, molly12, lyx324521, LQ1206, gitcode-chenjiao, cai-weiwei1989, CHDong, ogqin, yuanlipingGit, xuqinglin1, lqz2, zouwei1, chaoluoa, paradox325, jackzhang1116, yaoyao, akh, yujiacheng, dengjie0116, Hubert11111, Shine_Ws, wslhj555, longqiand, OYtao666, JiaqingQiang, luyyyy, Kingbelial, zhanghaiyu0101, wenxp1018, yanliu-luoluo, ksun_sekiro, liyong328, wgzheng, tangky, vivi_is_coding, aoiaoisola, weixin_44494597, wangmengmengwang65667, hid57809721, qq_35468730, comeonup, C547032, gcw_m5OQChA4, yao_yao_ling_xian, cnnbwcy, szqfes_12, cora_19, cann_lilin, can, shawnylee233]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
ascend-robot
ascend-robot成员
7月2日 评论:
流水线 PR-pipeline_pytorch#40695 [ commitID:312390ac ] 已完成
>>>代码风格自动修复执行成功(无修复内容)
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_LibTorch_ARM >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
patch_test 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
CodeCheck_lintrunner >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
UT_ARM_A2_Part_03 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_DIST_ARM_Part_01 🛑 >>>
UT_DIST_ARM_Part_02 🛑 >>>
UT_DIST_ARM_Part_03 🛑 >>>
UT_DIST_ARM_Part_04 🛑 >>>
UT_ARM_A2_Select_Part_01 >>>
UT_ARM_A2_Select_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
此流水线已支持下列评论快捷指令,仅PR创建者和白名单成员[wujinyuan1, huangjingwei, liangsongwei, yashi999, culechan, Dring, wuyouqi1, L1919_snow, qq_52711437, WhiteNight12, nomiz, xiu_21, ffmh, wanglijun55, hss-shuai, husichao, smallsilly, lanshaozuishuai, jimmyisme1, lzy0920232, alpha-junh, Sunshine_Youngster, wei_zhuoyi, zhangyihuiben, zyw-hw, zzzkeke, rmch, yangch0324, LucciC, AACAES, renyujin, wjlflyer, senzhen-town, pengjingyou, qsc97, limuan, yule100, xiaoqi-zhou, kuhn7, chenxingying, hanye02, zichun_ye, anyrenwei, kkjocker, wangzili121, Lu_G, yvjc, puddingfjz, HandsoemLemon, bigprestigee1, huawuyi, zhenyu10, dairenjie, du-jin-hang, zou-jieyu, adelaideliu, TrHan, wanlinan, Windwindzzz, pengqihw, kisnwang, yuheng_wang, honghao_wang, jizewei, zhangguoguang, sunyu-xuan, chenrayray, hbhu_bin, liujunzhu, c_34, LiNuoh, maoyuanpeng1, zzhongmin, zhaoyu65, bellatan, jiabaolin, zhuofanshen, wencaiwen, lu_zhuge, caoshuyang, molly12, lyx324521, LQ1206, gitcode-chenjiao, cai-weiwei1989, CHDong, ogqin, yuanlipingGit, xuqinglin1, lqz2, zouwei1, chaoluoa, paradox325, jackzhang1116, yaoyao, akh, yujiacheng, dengjie0116, Hubert11111, Shine_Ws, wslhj555, longqiand, OYtao666, JiaqingQiang, luyyyy, Kingbelial, zhanghaiyu0101, wenxp1018, yanliu-luoluo, ksun_sekiro, liyong328, wgzheng, tangky, vivi_is_coding, aoiaoisola, weixin_44494597, wangmengmengwang65667, hid57809721, qq_35468730, comeonup, C547032, gcw_m5OQChA4, yao_yao_ling_xian, cnnbwcy, szqfes_12, cora_19, cann_lilin, can, shawnylee233]评论有效
  • compile、compile_inductor、compile_torchair : 运行流水线
  • retry : 重试流水线所有失败子任务
  • retry <任务名> : 仅重试指定失败子任务
  • stop : 停止流水线
likedislike
atomgit-bot
atomgit-bot
7月2日 评论:

变更摘要

此 PR 修复了 profiler 在多卡分析场景下,由于配置缓存未正确重置导致的问题。通过在 ProfilerConfig 类中新增 reset_is_load() 方法,并在 ProfilingParser 解析流程中调用该方法,确保每次解析前 _is_load 标记被重置为 False,避免上一次加载的缓存配置影响后续多卡的分析结果。

主要改动

  • 新增 ProfilerConfig.reset_is_load() 方法:在 _profiler_config.py 中为 ProfilerConfig 类添加 reset_is_load() 方法,用于将 _is_load 标记重置为 False,提供清除配置加载缓存的能力。
  • 解析流程中增加缓存重置调用:在 _profiling_parser.pyProfilingParser 解析逻辑中,于调用 ProfilerConfig().load_info() 之前先执行 ProfilerConfig().reset_is_load(),确保每次解析多卡数据时配置加载状态是干净的。
  • 新增对应的单元测试:在 test_profiler_config.py 中添加 test_reset_is_load_will_reset_is_load_to_false 测试用例,验证 reset_is_load() 能将 _is_load 置为 False,同时不影响 _rank_id 等其他字段。
likedislike
atomgit-bot
atomgit-bot
7月2日 评论:

代码审查

经过对三个变更文件的逐一审查,包括对 Singleton 模式、load_info 缓存逻辑、调用链(analyse_profiling_datarun_parser → 各下游解析器)的完整上下文分析,未发现此 diff 引入任何正确性、安全性、可靠性或破坏性变更问题。

变更要点及验证结论:

  1. torch_npu/profiler/analysis/_profiler_config.py:新增 reset_is_load() 方法,将 _is_load 置为 False。实现简洁正确,ProfilerConfig@Singleton,方法仅操作实例自身状态,无副作用。

  2. torch_npu/profiler/analysis/_profiling_parser.py:在 analyse_profiling_data 中,于 load_info 之前调用 reset_is_load()。由于 ProfilerConfig 是单例,两次 ProfilerConfig() 返回同一实例,确保 _is_load 被重置后再加载配置。这解决了原 bug(多卡分析时配置缓存导致 rank 不一致),且与 no_exception_func 装饰器的异常处理逻辑兼容。

  3. test/profiler/analysis/test_profiler_config.py:新增 test_reset_is_load_will_reset_is_load_to_false 测试用例,验证 reset_is_load 仅重置 _is_load 而不影响 _rank_id 等其他状态。测试逻辑正确。


审查总结

  • P0/P1/P2 问题:0 个
  • P3 问题:0 个
  • 整体风险评估:此变更风险低,修复针对性强,无引入新缺陷的迹象。

各文件审查结果:

  • test/profiler/analysis/test_profiler_config.py — 无问题
  • torch_npu/profiler/analysis/_profiler_config.py — 无问题
  • torch_npu/profiler/analysis/_profiling_parser.py — 无问题

⚠️ 已识别出整体风险,但无法提取行内评论,请参考整体评估。

likedislike
renyujin成员
7月2日 评论:

/lgtm

likedislike
chenhao_1209成员
7月3日 评论:

/approve

likedislike
ascend-robotascend-robot成员
7月3日 添加了label:approvedlgtm
ascend-robotascend-robot成员
7月3日 合入了pull request
ascend-robot
ascend-robot成员
7月3日 评论:
流水线 pytorch_gitcode_PR_multiVersion#11782 [ commitID:312390ac ] 已完成
likedislike