Browse Source

fix all reduce and loss overlap

pull/315/head
baker 4 years ago
parent
commit
ef96a49483
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      ge/graph/build/logical_stream_allocator.cc

+ 2
- 0
ge/graph/build/logical_stream_allocator.cc View File

@@ -649,12 +649,14 @@ Status LogicalStreamAllocator::RunPasses(const ComputeGraphPtr &graph, const vec
if (context_.enable_single_stream) {
passes.emplace_back(MakeShared<SingleStreamPass>());
passes.emplace_back(MakeShared<NodeStreamUpdatePass>());
passes.emplace_back(MakeShared<UpdateForSkippedEnginePass>());
} else {
passes.emplace_back(MakeShared<AssignByLabelPass>());
passes.emplace_back(MakeShared<IndependentStreamPass>());
passes.emplace_back(MakeShared<AssignByDependencyPass>());
passes.emplace_back(MakeShared<NodeStreamUpdatePass>());
passes.emplace_back(MakeShared<AllReduceParallelPass>());
passes.emplace_back(MakeShared<UpdateForSkippedEnginePass>());
}

for (auto &pass : passes) {


Loading…
Cancel
Save