From 0f8b045a5f6d32af57b5cc20ba334bed9f4d3d39 Mon Sep 17 00:00:00 2001 From: isaactalx Date: Wed, 24 Feb 2021 16:38:50 +0800 Subject: [PATCH] static warning clean --- ge/hybrid/executor/hybrid_model_async_executor.cc | 3 ++- ge/hybrid/executor/hybrid_model_pipeline_executor.cc | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/ge/hybrid/executor/hybrid_model_async_executor.cc b/ge/hybrid/executor/hybrid_model_async_executor.cc index 1b5afb83..97fb9d50 100644 --- a/ge/hybrid/executor/hybrid_model_async_executor.cc +++ b/ge/hybrid/executor/hybrid_model_async_executor.cc @@ -26,6 +26,7 @@ namespace hybrid { namespace { const int kDataOutputIndex = 0; const size_t kMinimumPiplineStages = 2; +const int kDefaultLoopCount = 10; } HybridModelAsyncExecutor::HybridModelAsyncExecutor(HybridModel *model) : model_(model), run_flag_(false) { @@ -150,7 +151,7 @@ Status HybridModelAsyncExecutor::RunInternal() { GELOGI("HybridModel will execute in pipeline mode"); auto iter_per_run = std::getenv("ITER_NUM"); if (iter_per_run) { - args.num_loops = static_cast(strtol(iter_per_run, nullptr, 10)); + args.num_loops = static_cast(strtol(iter_per_run, nullptr, kDefaultLoopCount)); } ret = pipe_executor_->Execute(args); } else { diff --git a/ge/hybrid/executor/hybrid_model_pipeline_executor.cc b/ge/hybrid/executor/hybrid_model_pipeline_executor.cc index b043ca7f..6c824bf8 100644 --- a/ge/hybrid/executor/hybrid_model_pipeline_executor.cc +++ b/ge/hybrid/executor/hybrid_model_pipeline_executor.cc @@ -8,6 +8,7 @@ namespace ge { namespace hybrid { namespace { constexpr int kNumExecutors = 2; +const int kMinLoopCount = 2; const int kIntBase = 10; const char *const kEnvProfilingLevel = "HYBRID_PROFILING_LEVEL"; } @@ -208,7 +209,7 @@ Status HybridModelPipelineExecutor::InitStageExecutors() { Status HybridModelPipelineExecutor::Execute(HybridModelExecutor::ExecuteArgs &args) { int loop_count = args.num_loops; - GE_CHECK_GE(loop_count, 2); + GE_CHECK_GE(loop_count, kMinLoopCount); auto &inputs = args.inputs; auto &input_desc = args.input_desc;