From dcb91b0ef6a530d2ba5158b469498f0aed6a8894 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Thu, 28 May 2020 14:26:12 +0800 Subject: [PATCH] fix reshape reshape case in auto parallel --- mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h b/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h index 5e7ce7cdd..f1f73de4d 100644 --- a/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h +++ b/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h @@ -88,7 +88,9 @@ class TwoReshapeEliminater : public AnfVisitor { auto fg = node->func_graph(); if (fg != nullptr && x_ != nullptr && shape_ != nullptr) { - return fg->NewCNode({NewValueNode(prim_), x_, shape_}); + auto new_node = fg->NewCNode({NewValueNode(prim_), x_, shape_}); + new_node->set_abstract(node->abstract()); + return new_node; } return nullptr; } -- GitLab