已合并
change force_simt_only to is_pure_simt #45360
pangjiayi创建于 5 天前
change force_simt_only to is_pure_simt #45360
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 pangjiayi 的贡献)ascend-robot
5 天前 评论:
5 天前 评论:
atomgit-bot
5 天前 评论:
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_addr、sync_block_lock声明/赋值逻辑判断同步更新。- 运行时元数据键重命名:
triton_heuristics.py中NPUCachingAutotuner与NPUSymbolicGroupedAutotuner读取的元数据键由force_simt_only改为is_pure_simt,enable_simt的计算逻辑随之更新。 - fast-launch 绑定接口重命名:
bindings.cpp中 pybind 接口入参force_simt_only改为is_pure_simt,并同步更新注释与TORCH_CHECK报错文案;C++ 内部字段plan.forceSimtOnly保持不变。 - fast-launch Python 侧适配:
launcher.py与backend.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。


atomgit-bot
5 天前 评论:
5 天前 评论:
5 天前 添加了label:ascend-cla/yes
此处折叠了109条消息 查看更多
5 天前 修改了pull request 的描述
5 天前 删除了label:needs-issue
5 天前 合入了pull request
ascend-robot
5 天前 评论:
5 天前 评论:
Pull Request 已合并或已关闭。
If you want to solve this problem, you can click here to do it in the FAQs.


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


【合入来源】
issue #4391
【修改方案】
【资料变更】
【接口变更】
【功能验证】
【CheckList】