Browse Source

!1547 [Auto parallel] Fix some codestyle warnings

Merge pull request !1547 from Xiaoda/fix-some-rec-codestyle-warnings
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
c8d1f3f5be
2 changed files with 3 additions and 1 deletions
  1. +0
    -1
      mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc
  2. +3
    -0
      mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc

+ 0
- 1
mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_generate_strategy.cc View File

@@ -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

+ 3
- 0
mindspore/ccsrc/parallel/auto_parallel/rec_core/rec_partition.cc View File

@@ -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;


Loading…
Cancel
Save