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