已合并
add t5 config #44029
lihui创建于 11 天前
add t5 config #44029
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 lihui 的贡献)atomgit-bot
11 天前 评论:
11 天前 评论:
变更摘要
此 PR 将 native_dropout 操作的副作用注册(_register_effectful_op)从全局的 torch_npu/_inductor/dvm/decomp.py 迁移到 benchmarks/torchbench/npu_support.py 中 T5 模型的专用补丁函数 _patch_model_24 内,缩小了该注册的影响范围,使其仅在 T5 模型的 benchmark 测试场景下生效。
主要改动
native_dropout副作用注册位置迁移:将_register_effectful_op(torch.ops.aten.native_dropout.default, _EffectType.ORDERED)从torch_npu/_inductor/dvm/decomp.py的全局初始化代码中移除,改为在benchmarks/torchbench/npu_support.py的_patch_model_24函数内执行,使该注册仅对 T5 模型生效。- 移除
decomp.py中的相关导入:删除了torch_npu/_inductor/dvm/decomp.py中从torch._higher_order_ops.effects导入的_EffectType和_register_effectful_op,避免全局引入不必要的副作用注册。


ascend-robot
11 天前 评论:
11 天前 评论:
atomgit-bot
11 天前 评论:
11 天前 评论:
11 天前 添加了label:ascend-cla/yes
此处折叠了48条消息 查看更多
daifu1234567
7 天前 评论:
7 天前 评论:
/lgtm


7 天前 添加了label:lgtm
7 天前 删除了label:ci-pipeline-passed
7 天前 合入了pull request
ascend-robot
7 天前 评论:
7 天前 评论:
流水线 pytorch_gitcode_PR_multiVersion#13850 [ commitID:4acbc06b ] 运行失败


【合入来源】
【修改方案】
在npu_support.py里面针对T5网络增加dropout算子保序副作用
【资料变更】
否
【接口变更】
不涉及
【功能验证】
本地验证通过
【CheckList】