diff --git a/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc b/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc index 32172a3bb73d6b89969d38b923b32f730f23dfbf..b69fb20bee3a8a13260a304442a345c1263dfc8e 100644 --- a/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc +++ b/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc @@ -53,7 +53,9 @@ AnfNodePtr ArithmeticSimplify::operator()(const OptimizerPtr &, const AnfNodePtr auto ttmul = NewCNode({mul_node, const_.GetNode(node), const_2.GetNode(node)}, node->func_graph()); return NewCNode({mul_node, x.GetNode(node), ttmul}, node->func_graph()); } - return NewCNode({mul_node, x.GetNode(node), new_mul_tensor}, node->func_graph()); + auto new_cnode = NewCNode({mul_node, x.GetNode(node), new_mul_tensor}, node->func_graph()); + new_cnode->set_abstract(node->abstract()); + return new_cnode; }; MATCH_REPLACE_LAMBDA(node, const_ * (const_2 * x), const_dup_lambda);