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

!46940 [PyNative][Bugfix] Input to attr. 

Merge pull request !46940 from caifubi/r2.0.0-alpha-input-to-attr
This commit is contained in:
i-robot 2022-12-17 09:07:45 +00:00 committed by Gitee
parent 53ce6bf714 adfd1f8cd3
commit ab0ef9e32b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ascend_vm_op_adapter.cc
View File

@ -31,6 +31,13 @@ const AnfNodePtr AscendVmOpAdapter::Process(const FuncGraphPtr &graph, const Anf
if (node == nullptr || !AnfUtils::IsRealCNodeKernel(node)) { if (node == nullptr || !AnfUtils::IsRealCNodeKernel(node)) {
return nullptr; return nullptr;
} }
const auto kAttrAlreadyInputToAttr = "already_input_to_attr";
if (common::AnfAlgo::HasNodeAttr(kAttrAlreadyInputToAttr, node->cast<CNodePtr>())) {
return nullptr;
}
common::AnfAlgo::SetNodeAttr(kAttrAlreadyInputToAttr, MakeValue(true), node);
auto op_name = common::AnfAlgo::GetCNodeName(node); auto op_name = common::AnfAlgo::GetCNodeName(node);
auto is_dynamic = common::AnfAlgo::IsDynamicShape(node) || graph->has_flag(kAttrMutableKernel); auto is_dynamic = common::AnfAlgo::IsDynamicShape(node) || graph->has_flag(kAttrMutableKernel);
auto op_adaptation_info = auto op_adaptation_info =