使能二进制算子后报错“DynamicGRUV2GradAlignFusionPass cannot be applied for unknown shape”
2024/02/21
50
问题信息
问题来源 | 产品大类 | 产品子类 | 关键字 |
---|---|---|---|
官方 | 模型训练 | PyTorch | -- |
问题现象描述
- 硬件配置:
- CANN 7.0.RC1.alpha002
- PyTorch 1.8.1
- Python 3.7.5
- 源码地址及代码修改:https://github.com/Anery/RSAN。
- 问题现象:
将PyTorch框架RSAN模型迁移至NPU上发现训练性能很差,使能二进制算子后报错“op[DynamicGRUV2GradAlign], DynamicGRUV2GradAlignFusionPass cannot be applied for unknown shape.[FUNC:Fusion][FILE:dynamic_gru_v2_grad_align_fusion_pass.cc][LINE:1515]”,如下图所示:
原因分析
走二进制算子的DynamicGRUV2GradAlign等算子不支持动态shape。
解决措施
将这些算子添加至黑名单,使这些算子不走二进制,添加方法如下:
本页内容