From 9592852927ce64446e5ab0a3d53041fcbfb9c581 Mon Sep 17 00:00:00 2001
From: Xiaoda Zhang <zhangxiaoda@huawei.com>
Date: Thu, 28 May 2020 09:26:54 +0800
Subject: [PATCH] fix some codestyle warnings

---
 .../parallel/auto_parallel/rec_core/rec_generate_strategy.cc   | 1 -
 .../ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc     | 3 +++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc b/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
index 8c99df8345..5bccf73fc2 100644
--- a/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
+++ b/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
@@ -611,6 +611,5 @@ void GenerateEliminatedOperatorStrategyBackward(const std::vector<std::shared_pt
     ops[iter_ops]->SetSelectedStrategyAndCost(sp, ops[iter_ops]->selected_cost());
   }
 }
-
 }  // namespace parallel
 }  // namespace mindspore
diff --git a/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc b/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc
index ac8e52eed6..186987c0dd 100644
--- a/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc
+++ b/mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc
@@ -259,6 +259,9 @@ Graph::NodeType ApplyStrToTensor(Graph::NodeType Node) {
 
 Status DevicesMemoryControl(const size_t num_device, const double device_memory, std::shared_ptr<Graph> graph) {
   MS_EXCEPTION_IF_NULL(graph);
+  if (num_device == 0) {
+    MS_LOG(EXCEPTION) << "Failure: device number is 0.";
+  }
 
   uint64_t iter_nodes = graph->nodes.size();
   double used_memory = 0.0;