mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

dynamic_aicpu_kernel_mod should update output_size_list_ in Resize even 

when type is kComputeDepend
This commit is contained in:
ckey_Dou 2022-06-30 20:32:26 +08:00
parent 7e7cbe55a4
commit 506a18e039
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/dynamic_aicpu_kernel_mod.cc
View File

@ -62,10 +62,9 @@ int DynamicAicpuOpKernelMod::Resize(const BaseOperatorPtr &base_operator, const
if (!common::AnfAlgo::IsDynamicShape(cnode)) {
MS_LOG(EXCEPTION) << "The node is not dynamic shape: " << cnode->fullname_with_scope();
}
// update output size after InferShape.
if (unknow_type_ != device::ascend::UnknowShapeOpType::DEPEND_COMPUTE) {
AscendKernelMod::UpdateOutputSizeList();
}
AscendKernelMod::UpdateOutputSizeList();
MS_LOG(INFO) << "UpdateExtInfo of " << cnode->fullname_with_scope() << " start";
auto input_num = common::AnfAlgo::GetInputTensorNum(cnode);