使能二进制算子后报错“DynamicGRUV2GradAlignFusionPass cannot be applied for unknown shape”
更新时间： 2024/02/21
问题信息
| 问题来源 | 产品大类 | 产品子类 | 关键字 |
|---|---|---|---|
| 官方 | 模型训练 | 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。
解决措施
将这些算子添加至黑名单，使这些算子不走二进制，添加方法如下:
本页内容