From 06f43572fb37f3525c36b00d36bae0ffeb641c8c Mon Sep 17 00:00:00 2001
From: Giancarlo Colmenares <giancarlo.colmenares@huawei.com>
Date: Tue, 18 Aug 2020 10:44:02 -0400
Subject: [PATCH] Moved const multiply to ArithmeticSimplify2 to solve issue
 with auto-parallel failing because the shape is not yet set. Issue #67

---
 .../ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc    | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc b/mindspore/ccsrc/frontend/optimizer/irpass/arithmetic_simplify.cc
index 32172a3bb7..b69fb20bee 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);