2.4K Star 8.2K Fork 4.4K

GVPMindSpore / mindspore

 / 详情

【MS】【Lite】在mslite-2.2.12上跑动态推理时报错OpName:[Default/Gather-op3187] "batch_dims must be less than rank(params)1"

DONE
Bug-Report 成员
创建于  
2024-03-18 20:51

[ERROR] OP(3125924,python):2024-03-18-19:17:51.692.655 [selection_ops.cc:1171][OP_PROTO][GatherV2InferShape][3127658] OpName:[Default/Gather-op3187] "batch_dims must be less than rank(params)1

评论 (7)

Lin 创建了Bug-Report
Lin 添加了
 
v2.2.13
标签
Lin 添加了
 
mindspore-assistant
标签
Lin 添加了
 
kind/bug
标签
Lin 添加了
 
mindspore
标签
Lin 添加了
 
mindspore-lite
标签
展开全部操作日志

Please assign maintainer to check this issue.
请为此issue分配处理人。
@Lin

感谢您的提问,您可以评论//mindspore-assistant更快获取帮助:

  1. 如果您刚刚接触MindSpore,或许您可以在教程找到答案
  2. 如果您是资深Pytorch用户,您或许需要:
  1. 如果您遇到动态图问题,可以设置set_context(pynative_synchronize=True)查看报错栈协助定位
  2. 模型精度调优问题可参考官网调优指南
  3. 如果您反馈的是框架BUG,请确认您在ISSUE中提供了MindSpore版本、使用的后端类型(CPU、GPU、Ascend)、环境、训练的代码官方链接以及可以复现报错的代码的启动方式等必要的定位信息
  4. 如果您已经定位出问题根因,欢迎提交PR参与MindSpore开源社区,我们会尽快review
Lin 添加协作者Yang Guilong
Lin 计划开始日期2024-03-18 修改为
Lin 计划截止日期2024-03-20 修改为
Lin 计划开始日期设置为2024-03-18
Lin 计划截止日期设置为2024-03-20
Yang Guilong 修改了标题

补充说明:

开启双动态后导出mindir,然后运行推理时出现错误:

[ERROR] OP(3125924,python):2024-03-18-19:17:51.692.655 [selection_ops.cc:1171][OP_PROTO][GatherV2InferShape][3127658] OpName:[Default/Gather-op3187] "batch_dims must be less than rank(params)1

报错出现得较早,还没有走到编译prompt_kv_cache等算子那个环节。

不开启双动态没有这个错误。

用 mindspore_lite 2.2.12和2.2.13均有这个问题。但 2.2.11.20240123 则没有这个错误,能够正常跑通双动态推理

加了reshape后没问题

fangzehua 里程碑B-SIG-MSLite 修改为B-SIG-MindFormers
fangzehua 添加协作者fangzehua
fangzehua 负责人fangzehua 修改为Yang Guilong
fangzehua 取消协作者Yang Guilong
fangzehua 任务状态TODO 修改为VALIDATION

核查模型网络代码后发现是隐藏层之后的rms_norm忘记reshape了,加了reshape之后可以正常跑通,已在mindspore_lite 2.2.13上验证过了。单子可以驳回。

核查模型网络代码后发现是隐藏层之后的rms_norm忘记reshape了,加了reshape之后可以正常跑通,已在mindspore_lite 2.2.13上验证过了。单子可以驳回。

@Yang Guilong 关联合入的修复PR

PR在 mindformers 库,此单在 mindspore 库,好像无法关联上。

https://gitee.com/mindspore/mindformers/pulls/2509

Lin 任务状态VALIDATION 修改为DONE
i-robot 添加了
 
gitee
标签
Xinrui Chen 里程碑B-SIG-MindFormers 修改为未设置
Shawny 移除了
 
mindspore-assistant
标签
Shawny 移除了
 
mindspore-assistant
标签

登录 后才可以发表评论

状态
负责人
项目
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
开始日期   -   截止日期
-
置顶选项
优先级
预计工期 (小时)
参与者(4)
7578154 fangzehua 1590581079
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore

搜索帮助

344bd9b3 5694891 D2dac590 5694891