Browse Source

!947 change mult batch to switchn

From: @jiming6
Reviewed-by: @ji_chen,@wangxiaotian22
Signed-off-by: @startzgf168,@ji_chen
pull/947/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
c14c6a87cb
1 changed files with 5 additions and 7 deletions
  1. +5
    -7
      ge/graph/preprocess/multi_batch_copy_graph.cc

+ 5
- 7
ge/graph/preprocess/multi_batch_copy_graph.cc View File

@@ -1692,13 +1692,11 @@ Status MultiBatchGraphCopyer::LinkToNodeOutBranch(const NodePtr &node) {
}

Status ProcessMultiBatch(ComputeGraphPtr &graph) {
if (GetLocalOmgContext().dynamic_node_type.empty()) {
const char *multi_batch_with_switchn = std::getenv("MULTI_BATCH_WITH_SWITCHN");
if (multi_batch_with_switchn == nullptr) {
PassManager pass_manager;
GE_CHK_STATUS_RET(pass_manager.AddPass("MultiBatchClonePass", new (std::nothrow) MultiBatchClonePass));
return pass_manager.Run(graph);
}
const char *multi_batch_with_case = std::getenv("MULTI_BATCH_WITH_CASE");
if (multi_batch_with_case != nullptr) {
PassManager pass_manager;
GE_CHK_STATUS_RET(pass_manager.AddPass("MultiBatchClonePass", new (std::nothrow) MultiBatchClonePass));
return pass_manager.Run(graph);
}

if (!GetLocalOmgContext().need_multi_batch) {


Loading…
Cancel
Save