Browse Source

modify initial session id of BuildModel

pull/2073/head
wuweikang 3 years ago
parent
commit
9c40e93526
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      ge/generator/ge_generator.cc

+ 3
- 2
ge/generator/ge_generator.cc View File

@@ -55,6 +55,7 @@ const int64_t kDynamicDimValue = -2;
const int kDefaultDeviceId = 0;
const int kDefaultJobId = 0;
const int32_t kFuzzBuildPattern = 1;
const uint64_t kInitialSessionId = INT32_MAX;

std::map<ge::OpEngineType, std::string> engine_type_map{
{ge::ENGINE_SYS, kEngineNameDefault},
@@ -1137,8 +1138,8 @@ Status GeGenerator::Impl::BuildModel(const Graph &graph, const vector<GeTensor>
}

graph_manager_.SetOptionsRunGraphFlag(false);
static std::atomic<uint64_t> atomic_session_id(0);
// prevent from repeatation of session_id in mix using singleop and graph mode.
static std::atomic<uint64_t> atomic_session_id(kInitialSessionId);
auto session_id = atomic_session_id.fetch_add(1);
// This is a temporary add for graph with variable
auto version = static_cast<int32_t>(SessionVersion::ClOUD_VERSION);


Loading…
Cancel
Save