p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!24313 Process is stuck when update aicpu op shape failed 

Merge pull request !24313 from caifubi/master-fix-aicpu-ds
This commit is contained in:
i-robot 2021-09-28 11:05:14 +00:00 committed by Gitee
parent 45a6a571ae 26ac255b78
commit 36c591b113
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ccsrc/runtime/device/ascend/executor/ai_cpu_dynamic_kernel.cc
View File

@ -153,12 +153,18 @@ bool AiCpuDynamicKernel::UpdateExtInfo() {
MS_EXCEPTION_IF_NULL(ext_info_handler_);
for (size_t i = 0; i < input_num_; ++i) {
ext_info_handler_->UpdateInputShapeAndType(i, NOT_NULL(cnode));
if (!ext_info_handler_->UpdateInputShapeAndType(i, NOT_NULL(cnode))) {
MS_LOG(ERROR) << "Update input shape failed, cnode:" << cnode->fullname_with_scope() << " input:" << i;
return false;
}
}
if (AnfAlgo::IsDynamicShape(cnode) && unknow_type_ != DEPEND_COMPUTE) {
for (size_t i = 0; i < output_num_; ++i) {
ext_info_handler_->UpdateOutputShapeAndType(i, NOT_NULL(cnode));
if (!ext_info_handler_->UpdateOutputShapeAndType(i, NOT_NULL(cnode))) {
MS_LOG(ERROR) << "Update output shape failed, cnode:" << cnode->fullname_with_scope() << " output:" << i;
return false;
}
}
}