case plugin

pull/704/head
wjm 4 years ago
parent 4f97bc9abd
commit f91fdad16a

@ -1698,7 +1698,7 @@ Status ProcessMultiBatch(ComputeGraphPtr &graph) {
if (multi_batch_with_switchn == nullptr) { if (multi_batch_with_switchn == nullptr) {
PassManager pass_manager; PassManager pass_manager;
GE_CHK_STATUS_RET(pass_manager.AddPass("MultiBatchClonePass", new (std::nothrow) MultiBatchClonePass)); GE_CHK_STATUS_RET(pass_manager.AddPass("MultiBatchClonePass", new (std::nothrow) MultiBatchClonePass));
GE_CHK_STATUS_RET(pass_manager.AddPass("DataPass", new (std::nothrow) DataPass)); // 子图分配parent index GE_CHK_STATUS_RET(pass_manager.AddPass("DataPass", new (std::nothrow) DataPass)); // set subgraph parent index
return pass_manager.Run(graph); return pass_manager.Run(graph);
} }
} }

Loading…
Cancel
Save