From 49aacf9e6f76c8b9aa1c1b96b37d4df39cd36faf Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Sat, 13 Mar 2021 11:31:15 +0800 Subject: [PATCH] add for optune dynamic shape support --- ge/generator/ge_generator.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 515e42cb..568230cf 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -50,6 +50,8 @@ const char *const kFileNameSuffix = "online"; const char *const kAicpuAllshape = "_AllShape"; constexpr char const *kAttrSupportDynamicShape = "support_dynamicshape"; const int64_t kDynamicDimValue = -2; +const int kDefaultDeviceId = 0; +const int kDefaultJobId = 0; std::map engine_type_map{ {ge::ENGINE_SYS, kEngineNameDefault}, @@ -919,12 +921,10 @@ Status GeGenerator::Impl::BuildModel(const Graph &graph, const vector auto session_id = atomic_session_id.fetch_add(1); // This is a temporary add for graph with variable auto version = static_cast(SessionVersion::ClOUD_VERSION); - const int DEFAULT_DEVICE_ID = 0; - const int DEFAULT_JOB_ID= 0; - ret = VarManager::Instance(session_id)->Init(version, session_id, DEFAULT_DEVICE_ID, DEFAULT_JOB_ID); + ret = VarManager::Instance(session_id)->Init(version, session_id, kDefaultDeviceId, kDefaultJobId); GELOGI("Start init var instance, session_id %lu", session_id); if (ret != SUCCESS) { - GELOGE(ret, "Failed init var instance, session_id %lu", session_id); + GELOGW("Failed init var instance, session_id %lu", session_id); } if (is_singleop_unregistered_) { ret = graph_manager_.BuildGraphForUnregisteredOp(graph_id, inputs, ge_root_model, session_id);