fix reshape reshape case in auto parallel

dcb91b0e · yao_yf · 0f221403 · dcb91b0e
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h +3 -1

未找到文件。
--- 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;
  }