From 9a5e6d7e038c7029f5a4b4da581ea5e6ae311869 Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Fri, 7 May 2021 10:55:21 +0800 Subject: [PATCH] add support for train_mode tune --- ge/graph/passes/global_step_insert_pass.cc | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ge/graph/passes/global_step_insert_pass.cc b/ge/graph/passes/global_step_insert_pass.cc index 6ed7a7ec..d702e758 100755 --- a/ge/graph/passes/global_step_insert_pass.cc +++ b/ge/graph/passes/global_step_insert_pass.cc @@ -28,6 +28,10 @@ #include "graph/passes/pass_utils.h" #include "graph/ge_context.h" +namespace { +const char *const kFlagOff = "0"; +} // namespace + namespace ge { NodePtr GlobalStepInsertPass::InsertOp(ComputeGraphPtr &compute_graph, const string &node_type, @@ -73,8 +77,9 @@ NodePtr GlobalStepInsertPass::InsertOp(ComputeGraphPtr &compute_graph, } Status GlobalStepInsertPass::Run(ComputeGraphPtr compute_graph) { + // run_flag off means offline, no need insert global step node which type is variable std::string run_flag; - if (ge::GetContext().GetOption(ge::RUN_FLAG, run_flag) == GRAPH_SUCCESS && run_flag == "0") { + if (ge::GetContext().GetOption(ge::RUN_FLAG, run_flag) == GRAPH_SUCCESS && run_flag == kFlagOff) { GELOGI("compute_graph [%u] [%s] skip insert global step", compute_graph->GetGraphID(), compute_graph->GetName().c_str()); return SUCCESS;