已合并
change force_simt_only to is_pure_simt #45360
pangjiayi创建于 5 天前
change force_simt_only to is_pure_simt #45360
已合并
pangjiayi创建于 5 天前
pangjiayi
pangjiayi
5 天前

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

issue #4391

【修改方案】

请描述修改内容的具体实现,涉及哪些组件之间进行交互,可以用1、2、3、...进行罗列
如果是需求或者重构类的PR,需要补充详细设计文档(说明上下游组件关系、时序图、类图、DFX能力等内容)

【资料变更】

请确认是否涉及资料变更。如涉及,需要在PR中体现,并简要说明修改内容。如不涉及,需填写“不涉及”

【接口变更】

请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及,需要详细说明接口以及对应的变更内容,同时需要在资料中体现。如不涉及,需填写“不涉及”

【功能验证】

说明测试场景,测试方法。如果本次测试方式与常规单元测试不同,请详细说明您的测试步骤
新增/变更内容是否已新增/适配UT测试用例看护,并补充测试自验证截图

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 pangjiayi 的贡献)
pangjiayipangjiayi
5 天前 创建了 pull request,commit cd244f24
atomgit-bot
atomgit-bot
5 天前 评论:

变更摘要

本 PR 为一次命名重构,将 Ascend 950 上 Triton 内核 fast-launch / codegen 链路中表示"纯 SIMT 模式"的字段 force_simt_only 统一重命名为 is_pure_simt,覆盖元数据键、pybind 接口入参、局部变量及注释/错误提示,使命名更贴近其实际语义。共涉及 6 个文件,改动为等量的字段/参数改名(+23/-23),不含逻辑行为变更。

主要改动

  • cpp_wrapper_npu.py 代码生成适配: DeferredNpuTritonCallWrapper 中变量及 generate_args_decl 的方法签名参数由 force_simt_only 改为 is_pure_simt,相应的 workspace_addrsync_block_lock 声明/赋值逻辑判断同步更新。
  • 运行时元数据键重命名: triton_heuristics.pyNPUCachingAutotunerNPUSymbolicGroupedAutotuner 读取的元数据键由 force_simt_only 改为 is_pure_simtenable_simt 的计算逻辑随之更新。
  • fast-launch 绑定接口重命名: bindings.cpp 中 pybind 接口入参 force_simt_only 改为 is_pure_simt,并同步更新注释与 TORCH_CHECK 报错文案;C++ 内部字段 plan.forceSimtOnly 保持不变。
  • fast-launch Python 侧适配: launcher.pybackend.py 中局部变量及元数据取值由 force_simt_only 改为 is_pure_simt,但 launcher 属性 _npu_fast_launch_force_simt_only 保留原名,保证对外行为不变。
  • 测试用例同步更新: test_grouped_cpp_wrapper.py 中构建分组 wrapper 测试元数据字典的两处 force_simt_only 字段改为 is_pure_simt
likedislike
atomgit-bot
atomgit-bot
5 天前 评论:

代码审查

✅ 未发现问题

likedislike
ascend-robotascend-robot成员
5 天前 添加了label:ascend-cla/yes
此处折叠了109条消息 查看更多
pangjiayipangjiayi
5 天前 修改了pull request 的描述
ascend-robotascend-robot成员
5 天前 删除了label:needs-issue
ascend-robotascend-robot成员
5 天前 合入了pull request
ascend-robot
ascend-robot成员
5 天前 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike
ascend-robot
ascend-robot成员
5 天前 评论:
流水线 pytorch_gitcode_PR_multiVersion#14568 [ commitID:7a2766ba ] 已完成
likedislike