From a3906fb127569da247ee5bd512386463b5510e71 Mon Sep 17 00:00:00 2001 From: heyingjiao Date: Mon, 26 Feb 2024 17:07:10 +0800 Subject: [PATCH] =?UTF-8?q?concat=20=E5=8A=A8=E6=80=81shape=E9=97=AE?= =?UTF-8?q?=E9=A2=98=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../cloud_infer/models_mindir_cloud_ascend.cfg | 2 ++ .../converter/adapter/acl/mapper/spatial_node_adapter.cc | 5 ++++- mindspore/lite/tools/optimizer/graph/args_to_attr_pass.cc | 8 +++++--- 3 files changed, 11 insertions(+), 4 deletions(-) diff --git a/mindspore/lite/test/config_level0/cloud_infer/models_mindir_cloud_ascend.cfg b/mindspore/lite/test/config_level0/cloud_infer/models_mindir_cloud_ascend.cfg index 87a1d2fef9d..6a1dcfe9cbe 100644 --- a/mindspore/lite/test/config_level0/cloud_infer/models_mindir_cloud_ascend.cfg +++ b/mindspore/lite/test/config_level0/cloud_infer/models_mindir_cloud_ascend.cfg @@ -27,3 +27,5 @@ deeplabv3_vocaug_bs_1_ascend -1 resnet50_imagenet_bs_1_Ascend 3 resnet50_imagenet_bs_1_GPU 3 unet3d_luna16_bs_1_ascend +fasterrcnn_coco2017_bs_1_ascend -1 +lstm_aclimdb_bs_64_Ascend -1 diff --git a/mindspore/lite/tools/converter/adapter/acl/mapper/spatial_node_adapter.cc b/mindspore/lite/tools/converter/adapter/acl/mapper/spatial_node_adapter.cc index 35871270f70..d1b4c3a61e3 100644 --- a/mindspore/lite/tools/converter/adapter/acl/mapper/spatial_node_adapter.cc +++ b/mindspore/lite/tools/converter/adapter/acl/mapper/spatial_node_adapter.cc @@ -76,7 +76,10 @@ static STATUS AdapteNodeWithDynamicInput(const CNodePtr &cnode) { cnode->set_inputs(new_inputs); // add kAttrDynInputSizes for multi-input operator. int64_t input_num = tuple_node->size() - 1; - prim->AddAttr(kAttrDynInputSizes, MakeValue(std::vector{input_num, -1})); + auto dst_prim = prim->Clone(); + dst_prim->AddAttr(kAttrDynInputSizes, MakeValue(std::vector{input_num, -1})); + ValueNodePtr value_node = cnode->input(0)->cast(); + value_node->set_value(dst_prim); return lite::RET_OK; } diff --git a/mindspore/lite/tools/optimizer/graph/args_to_attr_pass.cc b/mindspore/lite/tools/optimizer/graph/args_to_attr_pass.cc index 7af09104c48..52936c8d265 100644 --- a/mindspore/lite/tools/optimizer/graph/args_to_attr_pass.cc +++ b/mindspore/lite/tools/optimizer/graph/args_to_attr_pass.cc @@ -46,7 +46,7 @@ bool ArgsToAttrPass::Run(const FuncGraphPtr &func_graph) { if (prim == nullptr) { continue; } - + auto dst_prim = prim->Clone(); auto node_inputs = cnode->inputs(); std::vector new_node_inputs; @@ -85,10 +85,12 @@ bool ArgsToAttrPass::Run(const FuncGraphPtr &func_graph) { } auto arg_value_node = arg_input_node->cast(); auto arg_value = arg_value_node->value(); - prim->AddAttr(arg.arg_name_, arg_value); + dst_prim->AddAttr(arg.arg_name_, arg_value); + ValueNodePtr value_node = cnode->input(0)->cast(); + value_node->set_value(dst_prim); } - auto new_node = func_graph->NewCNode(prim, new_node_inputs); + auto new_node = func_graph->NewCNode(dst_prim, new_node_inputs); new_node->set_abstract(node->abstract()); new_node->set_fullname_with_scope(node->fullname_with_scope());