From 91b81385c3f04adc6d8a4c75fcbfbed337a97733 Mon Sep 17 00:00:00 2001 From: Xiaoda Zhang Date: Thu, 28 May 2020 15:23:21 +0800 Subject: [PATCH] add edge-creating error information --- mindspore/ccsrc/parallel/auto_parallel/edge_costmodel.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/ccsrc/parallel/auto_parallel/edge_costmodel.cc b/mindspore/ccsrc/parallel/auto_parallel/edge_costmodel.cc index 16393eb1736..60256a3ae32 100644 --- a/mindspore/ccsrc/parallel/auto_parallel/edge_costmodel.cc +++ b/mindspore/ccsrc/parallel/auto_parallel/edge_costmodel.cc @@ -95,6 +95,11 @@ Status Edge::InitEdgeCost() { << " failed, it may be caused by setting 'elementwise_op_strategy_follow' true. " "Try to set 'elementwise_op_strategy_follow' false."; } + if (edge_name_.find(RESHAPE) != std::string::npos) { + MS_LOG(EXCEPTION) << "Generating cost for edge: " << edge_name_ + << " failed, it may be caused by setting different strategies for operators following Reshape. " + "Try to fix that."; + } MS_LOG(EXCEPTION) << "Generating cost for edge: " << edge_name_ << " failed."; } return Status::SUCCESS;