Browse Source

Pre Merge pull request !2073 from HW_KK/session_id

pull/2073/MERGE
HW_KK Gitee 3 years ago
parent
commit
5708271a1a
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

@@ -56,6 +56,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},
@@ -1138,8 +1139,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