You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

inner_session.cc 23 kB

5 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
4 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534
  1. /**
  2. * Copyright 2020 Huawei Technologies Co., Ltd
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #include "session/inner_session.h"
  17. #include <map>
  18. #include <memory>
  19. #include <vector>
  20. #include "analyzer/analyzer.h"
  21. #include "adx_datadump_server.h"
  22. #include "common/dump/dump_properties.h"
  23. #include "common/dump/dump_manager.h"
  24. #include "framework/common/util.h"
  25. #include "framework/common/debug/ge_log.h"
  26. #include "graph/ge_context.h"
  27. #include "graph/ge_global_options.h"
  28. #include "graph/ge_local_context.h"
  29. #include "graph/common/local_context.h"
  30. #include "graph/load/model_manager/model_manager.h"
  31. #include "graph/manager/graph_var_manager.h"
  32. #include "graph/manager/graph_mem_manager.h"
  33. #include "graph/utils/tensor_adapter.h"
  34. #include "runtime/mem.h"
  35. #include "ir_build/option_utils.h"
  36. namespace ge {
  37. namespace {
  38. const int32_t kDumpStatus = 0;
  39. Status CheckReuseMemoryOption(const std::map<string, string> &options) {
  40. auto iter = options.find(OPTION_EXEC_DISABLE_REUSED_MEMORY);
  41. if (iter != options.end()) {
  42. if (iter->second == "0") {
  43. GELOGD("%s=0, reuse memory is open", OPTION_EXEC_DISABLE_REUSED_MEMORY);
  44. } else if (iter->second == "1") {
  45. GELOGD("%s=1, reuse memory is close", OPTION_EXEC_DISABLE_REUSED_MEMORY);
  46. } else {
  47. GELOGE(PARAM_INVALID, "[CheckReuse][MemoryOption]option %s=%s is invalid",
  48. OPTION_EXEC_DISABLE_REUSED_MEMORY, iter->second.c_str());
  49. REPORT_INNER_ERROR("E19999", "CheckReuseMemoryOption failed because option %s=%s is invalid.",
  50. OPTION_EXEC_DISABLE_REUSED_MEMORY, iter->second.c_str());
  51. return FAILED;
  52. }
  53. }
  54. return SUCCESS;
  55. }
  56. }
  57. static std::mutex mutex_; // BuildGraph and RunGraph use
  58. bool InnerSession::is_dump_server_inited_ = false;
  59. InnerSession::InnerSession(uint64_t session_id, const std::map<string, string> &options)
  60. : init_flag_(false), session_id_(session_id), options_(options) {}
  61. Status InnerSession::Initialize() {
  62. if (init_flag_) {
  63. GELOGW("[InnerSession:%lu] session already initialize.", session_id_);
  64. return SUCCESS;
  65. }
  66. // If the global options and the session options are duplicated, the session options is preferred.
  67. auto all_options = options_;
  68. all_options.insert(GetMutableGlobalOptions().begin(), GetMutableGlobalOptions().end());
  69. Status ret = CheckReuseMemoryOption(all_options);
  70. if (ret != SUCCESS) {
  71. GELOGE(ret, "[CheckReuse][MemoryOption] failed, [InnerSession:%lu].", session_id_);
  72. REPORT_CALL_ERROR("E19999", "CheckReuseMemoryOption failed, InnerSession=%lu.", session_id_);
  73. return ret;
  74. }
  75. // Check option modify_mixlist
  76. if (ge::CheckModifyMixlistParamValid(all_options) != ge::SUCCESS) {
  77. return FAILED;
  78. }
  79. UpdateThreadContext(std::map<std::string, std::string>{});
  80. // session device id set here
  81. std::string str_session_device_id;
  82. if (GetContext().GetOption("ge.session_device_id", str_session_device_id) == SUCCESS) {
  83. GELOGI("Option session device id has set, value is %s.", str_session_device_id.c_str());
  84. uint32_t session_device_id = 0;
  85. try {
  86. session_device_id = static_cast<uint32_t>(std::stoi(str_session_device_id.c_str()));
  87. // session device id has priority
  88. GetContext().SetCtxDeviceId(session_device_id);
  89. } catch (std::invalid_argument &) {
  90. GELOGW("session device id %s transform to int failed.", str_session_device_id.c_str());
  91. } catch (std::out_of_range &) {
  92. GELOGW("session device id %s transform to int failed.", str_session_device_id.c_str());
  93. }
  94. }
  95. GE_CHK_RT_RET(rtSetDevice(GetContext().DeviceId()));
  96. DumpProperties dump_properties;
  97. GE_CHK_STATUS_RET(dump_properties.InitByOptions(), "Init dump properties failed.");
  98. GE_CHK_STATUS_RET(AddDumpProperties(dump_properties), "[Add][DumpProperties] failed.");
  99. ret = graph_manager_.Initialize(options_);
  100. if (ret != SUCCESS) {
  101. GELOGE(ret, "[Init][GraphManager] failed, InnerSession:%lu.", session_id_);
  102. REPORT_CALL_ERROR("E19999", "GraphManager initialize failed, InnerSession:%lu.", session_id_);
  103. GE_CHK_STATUS(RemoveDumpProperties(), "[Remove][DumpProperties] failed.");
  104. return ret;
  105. }
  106. ret = VarManager::Instance(session_id_)->SetMemoryMallocSize(all_options);
  107. if (ret != SUCCESS) {
  108. GELOGE(ret, "[Set][MemoryMallocSize] failed.");
  109. REPORT_CALL_ERROR("E19999", "VarManager SetMemoryMallocSize failed, InnerSession:%lu.", session_id_);
  110. (void)graph_manager_.Finalize();
  111. GE_CHK_STATUS(RemoveDumpProperties(), "[Remove][DumpProperties] failed.");
  112. GE_CHK_RT(rtDeviceReset(static_cast<int32_t>(GetContext().DeviceId())));
  113. return ret;
  114. }
  115. int32_t version = static_cast<int32_t>(SessionVersion::ClOUD_VERSION);
  116. const int DEFAULT_DEVICE_ID = 0;
  117. const int DEFAULT_JOB_ID = 0;
  118. ret = VarManager::Instance(session_id_)->Init(version, session_id_, DEFAULT_DEVICE_ID, DEFAULT_JOB_ID);
  119. if (ret != SUCCESS) {
  120. GELOGE(ret, "[Init][VarManager] failed.");
  121. REPORT_CALL_ERROR("E19999", "VarManager init failed, InnerSession:%lu.", session_id_);
  122. GE_CHK_STATUS(RemoveDumpProperties(), "[Remove][DumpProperties] failed.");
  123. }
  124. init_flag_ = true;
  125. return SUCCESS;
  126. }
  127. Status InnerSession::Finalize() {
  128. std::lock_guard<std::mutex> lock(resource_mutex_);
  129. if (!init_flag_) {
  130. GELOGW("[InnerSession:%lu] session does not initialize.", session_id_);
  131. return SUCCESS;
  132. }
  133. UpdateThreadContext(std::map<std::string, std::string>{});
  134. Status ret = graph_manager_.Finalize();
  135. if (ret != SUCCESS) {
  136. // Subsequent code execution is required, so no return is required
  137. GELOGE(ret, "[Finalize][GraphManager] failed, InnerSession:%lu.", session_id_);
  138. REPORT_CALL_ERROR("E19999", "GraphManager Finalize failed, InnerSession:%lu.", session_id_);
  139. }
  140. ModelManager::GetInstance()->DestroyAicpuSession(session_id_);
  141. init_flag_ = false;
  142. // release var memory
  143. GELOGI("VarManager free var memory.");
  144. (void)VarManager::Instance(session_id_)->FreeVarMemory();
  145. for (auto memory_type : MemManager::Instance().GetAllMemoryType()) {
  146. (void)MemManager::Instance().SessionScopeMemInstance(memory_type).Free(session_id_);
  147. }
  148. // release analyzer saved info(Session Level)
  149. Analyzer::GetInstance()->DestroySessionJsonObject(session_id_);
  150. GE_CHK_RT(rtDeviceReset(static_cast<int32_t>(GetContext().DeviceId())));
  151. GE_CHK_STATUS_RET(RemoveDumpProperties(), "[Remove][DumpProperties] failed.");
  152. return ret;
  153. }
  154. Status InnerSession::GetVariable(const std::string &name, Tensor &val) {
  155. UpdateThreadContext(std::map<std::string, std::string>{});
  156. return graph_manager_.GetVariable(name, val);
  157. }
  158. Status InnerSession::AddGraph(uint32_t graph_id, const Graph &graph) {
  159. std::map<std::string, std::string> options;
  160. return AddGraph(graph_id, graph, options);
  161. }
  162. Status InnerSession::AddGraph(uint32_t graph_id, const Graph &graph,
  163. const std::map<std::string, std::string> &options) {
  164. std::lock_guard<std::mutex> lock(resource_mutex_);
  165. if (!init_flag_) {
  166. GELOGE(GE_SESS_INIT_FAILED, "[Add][Graph] failed because GraphManager not init, InnerSession:%lu, graph_id:%u.",
  167. session_id_, graph_id);
  168. REPORT_INNER_ERROR("E19999", "AddGraph failed because GraphManager not init, InnerSession:%lu, graph_id:%u.",
  169. session_id_, graph_id);
  170. return GE_SESS_INIT_FAILED;
  171. }
  172. UpdateThreadContext(options);
  173. Status ret = graph_manager_.AddGraph(graph_id, graph, options, domi::GetContext());
  174. if (ret != SUCCESS) {
  175. GELOGE(ret, "[Add][Graph] failed, InnerSession:%lu graphid: %u.", session_id_, graph_id);
  176. REPORT_CALL_ERROR("E19999", "GraphManager AddGraph failed, InnerSession:%lu graphid: %u.", session_id_, graph_id);
  177. return ret;
  178. }
  179. GELOGI("[InnerSession:%lu] add graph success, graph_id=%u.", session_id_, graph_id);
  180. return SUCCESS;
  181. }
  182. Status InnerSession::AddGraphWithCopy(uint32_t graph_id, const Graph &graph,
  183. const std::map<std::string, std::string> &options) {
  184. std::lock_guard<std::mutex> lock(resource_mutex_);
  185. if (!init_flag_) {
  186. GELOGE(GE_SESS_INIT_FAILED, "[Add][Graph] failed because GraphManager not init, InnerSession:%lu, graph_id:%u.",
  187. session_id_, graph_id);
  188. REPORT_INNER_ERROR("E19999",
  189. "AddGraphWithCopy failed because GraphManager not init, InnerSession:%lu, graph_id:%u.",
  190. session_id_, graph_id);
  191. return GE_SESS_INIT_FAILED;
  192. }
  193. UpdateThreadContext(options);
  194. Status ret = graph_manager_.AddGraphWithCopy(graph_id, graph, options, domi::GetContext());
  195. if (ret != SUCCESS) {
  196. GELOGE(ret, "[Add][Graph] failed, InnerSession:%lu graphid: %u.", session_id_, graph_id);
  197. REPORT_CALL_ERROR("E19999",
  198. "GraphManager AddGraphWithCopy failed, InnerSession:%lu graphid: %u.", session_id_, graph_id);
  199. return ret;
  200. }
  201. GELOGI("[InnerSession:%lu] add graph success, graph_id=%u.", session_id_, graph_id);
  202. return SUCCESS;
  203. }
  204. Status InnerSession::RunGraph(uint32_t graph_id, const std::vector<Tensor> &inputs, std::vector<Tensor> &outputs) {
  205. GELOGI("[InnerSession:%lu] run graph on session, graph_id=%u.", session_id_, graph_id);
  206. if (mutex_.try_lock()) {
  207. std::lock_guard<std::mutex> lock(mutex_, std::adopt_lock);
  208. if (!init_flag_) {
  209. GELOGE(GE_SESS_INIT_FAILED, "[Run][Graph]failed because GraphManager not Init, InnerSession:%lu, graph_id:%u.",
  210. session_id_, graph_id);
  211. REPORT_INNER_ERROR("E19999", "RunGraph failed because GraphManager not Init, InnerSession:%lu, graph_id:%u.",
  212. session_id_, graph_id);
  213. return GE_SESS_INIT_FAILED;
  214. }
  215. UpdateThreadContext(graph_id);
  216. vector<GeTensor> geInputs;
  217. for (auto &item : inputs) {
  218. geInputs.push_back(TensorAdapter::AsGeTensor(item));
  219. }
  220. vector<GeTensor> geOutputs;
  221. Status ret = graph_manager_.RunGraph(graph_id, geInputs, geOutputs, session_id_);
  222. domi::GetContext().out_nodes_map.clear();
  223. domi::GetContext().user_out_nodes.clear();
  224. if (ret != SUCCESS) {
  225. GELOGE(ret, "[Run][Graph]failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  226. REPORT_CALL_ERROR("E19999",
  227. "GraphManager RunGraph failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  228. return ret;
  229. }
  230. outputs.clear();
  231. for (auto &item : geOutputs) {
  232. outputs.push_back(TensorAdapter::AsTensor(item));
  233. }
  234. GELOGI("[InnerSession:%lu] run graph success, graph_id=%u.", session_id_, graph_id);
  235. return SUCCESS;
  236. } else {
  237. GELOGE(GE_SESS_ALREADY_RUNNING, "[Run][Graph]failed, InnerSession:%lu, graph_id=%u.", session_id_, graph_id);
  238. REPORT_INNER_ERROR("E19999",
  239. "RunGraph failed because mutex try_lock false, InnerSession:%lu, graph_id=%u.",
  240. session_id_, graph_id);
  241. return GE_SESS_ALREADY_RUNNING;
  242. }
  243. }
  244. Status InnerSession::RunGraphWithStreamAsync(uint32_t graph_id, rtStream_t stream,
  245. const std::vector<Tensor> &inputs, std::vector<Tensor> &outputs) {
  246. GELOGI("Run graph with stream, session id = %lu, graph id = %u, stream = %p.",
  247. session_id_, graph_id, stream);
  248. if (mutex_.try_lock()) {
  249. std::lock_guard<std::mutex> lock(mutex_, std::adopt_lock);
  250. if (!init_flag_) {
  251. GELOGE(GE_SESS_INIT_FAILED, "[Run][GraphWithStream]failed because GraphManager not Init,"
  252. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  253. REPORT_INNER_ERROR("E19999", "RunGraphWithStreamAsync failed because GraphManager not Init,"
  254. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  255. return GE_SESS_INIT_FAILED;
  256. }
  257. UpdateThreadContext(graph_id);
  258. vector<GeTensor> ge_inputs;
  259. for (auto &item : inputs) {
  260. ge_inputs.emplace_back(TensorAdapter::AsGeTensor(item));
  261. }
  262. vector<GeTensor> ge_outputs;
  263. for (auto &item : outputs) {
  264. ge_outputs.emplace_back(TensorAdapter::AsGeTensor(item));
  265. }
  266. Status ret = graph_manager_.RunGraphWithStreamAsync(graph_id, stream, session_id_, ge_inputs, ge_outputs);
  267. domi::GetContext().out_nodes_map.clear();
  268. domi::GetContext().user_out_nodes.clear();
  269. if (ret != SUCCESS) {
  270. GELOGE(ret, "[Run][GraphWithStreamAsync]failed,"
  271. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  272. REPORT_CALL_ERROR("E19999", "GraphManager RunGrapWithStreamhAsync failed,"
  273. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  274. return ret;
  275. }
  276. GELOGI("Run graph with stream success, session id = %lu, graph id = %u, stream = %p.",
  277. session_id_, graph_id, stream);
  278. return SUCCESS;
  279. } else {
  280. GELOGE(GE_SESS_ALREADY_RUNNING, "[Run][GraphWithStreamAsync]failed because mutex try_lock false,"
  281. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  282. REPORT_INNER_ERROR("E19999", "[Run][GraphWithStreamAsync]failed failed because mutex try_lock false,"
  283. "session id = %lu, graph id = %u, stream = %p.", session_id_, graph_id, stream);
  284. return GE_SESS_ALREADY_RUNNING;
  285. }
  286. }
  287. Status InnerSession::RemoveGraph(uint32_t graph_id) {
  288. std::lock_guard<std::mutex> lock(resource_mutex_);
  289. if (!init_flag_) {
  290. GELOGE(GE_SESS_INIT_FAILED,
  291. "[Remove][Graph] failed because GraphManager not init, InnerSession:%lu, graph_id=%u.",
  292. session_id_, graph_id);
  293. REPORT_INNER_ERROR("E19999",
  294. "RemoveGraph failed, because GraphManager not init, InnerSession:%lu, graph_id=%u.",
  295. session_id_, graph_id);
  296. return GE_SESS_INIT_FAILED;
  297. }
  298. UpdateThreadContext(graph_id);
  299. Status ret = graph_manager_.RemoveGraph(graph_id);
  300. if (ret != SUCCESS) {
  301. GELOGE(ret, "[Remove][Graph] failed, InnerSession:%lu, graph_id=%u.", session_id_, graph_id);
  302. REPORT_CALL_ERROR("E19999",
  303. "GraphManager RemoveGraph failed, InnerSession:%lu, graph_id=%u.", session_id_, graph_id);
  304. return ret;
  305. }
  306. GELOGI("[InnerSession:%lu] remove graph success, graph_id=%u.", session_id_, graph_id);
  307. return SUCCESS;
  308. }
  309. Status InnerSession::RegisterCallBackFunc(
  310. const std::string &key,
  311. const std::function<Status(uint32_t, const std::map<std::string, ge::Tensor> &)> &callback) {
  312. std::lock_guard<std::mutex> lock(resource_mutex_);
  313. if (!init_flag_) {
  314. GELOGE(GE_SESS_INIT_FAILED,
  315. "[Register][CallBackFunc] failed because GraphManager not initialize, InnerSession:%lu.", session_id_);
  316. REPORT_INNER_ERROR("E19999",
  317. "RegisterCallBackFunc failed because GraphManager not init, InnerSession:%lu.", session_id_);
  318. return GE_SESS_INIT_FAILED;
  319. }
  320. UpdateThreadContext(std::map<std::string, std::string>{});
  321. Status ret = graph_manager_.RegisterCallBackFunc(key, callback);
  322. if (ret != SUCCESS) {
  323. GELOGE(ret, "[Register][CallBackFunc] failed, InnerSession:%lu register %s.", session_id_, key.c_str());
  324. REPORT_CALL_ERROR("E19999",
  325. "GraphManager RegisterCallBackFunc failed, InnerSession:%lu register %s.",
  326. session_id_, key.c_str());
  327. return ret;
  328. }
  329. GELOGI("[InnerSession:%lu] register %s callback function success.", session_id_, key.c_str());
  330. return SUCCESS;
  331. }
  332. Status InnerSession::RegisterCallBackFunc(
  333. const std::string &key,
  334. const std::function<Status(uint32_t, const std::map<AscendString, ge::Tensor> &)> &callback) {
  335. std::lock_guard<std::mutex> lock(resource_mutex_);
  336. if (!init_flag_) {
  337. GELOGE(GE_SESS_INIT_FAILED,
  338. "[Register][CallBackFunc]failed because GraphManager not initialize, InnerSession:%lu.", session_id_);
  339. REPORT_INNER_ERROR("E19999",
  340. "RegisterCallBackFunc failed because GraphManager not initialize, InnerSession:%lu.",
  341. session_id_);
  342. return GE_SESS_INIT_FAILED;
  343. }
  344. UpdateThreadContext(std::map<std::string, std::string>{});
  345. Status ret = graph_manager_.RegisterCallBackFunc(key, callback);
  346. if (ret != SUCCESS) {
  347. GELOGE(ret, "[Register][CallBackFunc] failed, InnerSession:%lu register %s.", session_id_, key.c_str());
  348. REPORT_CALL_ERROR("E19999",
  349. "GraphManager RegisterCallBackFunc failed, InnerSession:%lu register %s.",
  350. session_id_, key.c_str());
  351. return ret;
  352. }
  353. GELOGI("[InnerSession:%lu] register %s callback function success.", session_id_, key.c_str());
  354. return SUCCESS;
  355. }
  356. Status InnerSession::BuildGraph(uint32_t graph_id, const std::vector<InputTensorInfo> &inputs) {
  357. UpdateThreadContext(graph_id);
  358. GELOGI("[InnerSession:%lu] build graph on session, graph_id=%u.", session_id_, graph_id);
  359. std::vector<ge::GeTensor> ge_inputs;
  360. for (auto const &input : inputs) {
  361. std::vector<int64_t> input_dims;
  362. std::transform(input.dims.begin(), input.dims.end(), std::back_inserter(input_dims),
  363. [](int64_t x) -> int64_t { return x; });
  364. GeShape input_shape(input_dims);
  365. GeTensorDesc input_tensor_desc;
  366. input_tensor_desc.SetShape(input_shape);
  367. input_tensor_desc.SetDataType(static_cast<ge::DataType>(input.data_type));
  368. ge_inputs.emplace_back(input_tensor_desc);
  369. }
  370. GeRootModelPtr ge_root_model = nullptr;
  371. Status ret = graph_manager_.BuildGraph(graph_id, ge_inputs, ge_root_model, session_id_, true);
  372. if (ret != SUCCESS) {
  373. GELOGE(ret, "[Build][Graph] failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  374. REPORT_CALL_ERROR("E19999",
  375. "GraphManager BuildGraph failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  376. return ret;
  377. }
  378. GELOGI("[InnerSession:%lu] build graph success, graph_id=%u.", session_id_, graph_id);
  379. return ret;
  380. }
  381. Status InnerSession::BuildGraph(uint32_t graph_id, const std::vector<ge::Tensor> &inputs) {
  382. UpdateThreadContext(graph_id);
  383. GELOGI("[InnerSession:%lu] build graph on session, graph_id=%u.", session_id_, graph_id);
  384. std::vector<ge::GeTensor> ge_inputs;
  385. for (const auto &input : inputs) {
  386. ge_inputs.emplace_back(TensorAdapter::AsGeTensor(input));
  387. }
  388. GeRootModelPtr ge_root_model = nullptr;
  389. Status ret = graph_manager_.BuildGraph(graph_id, ge_inputs, ge_root_model, session_id_, true);
  390. if (ret != SUCCESS) {
  391. GELOGE(ret, "[Build][Graph] failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  392. REPORT_CALL_ERROR("E19999",
  393. "GraphManager BuildGraph failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  394. return ret;
  395. }
  396. GELOGI("[InnerSession:%lu] build graph success, graph_id=%u.", session_id_, graph_id);
  397. return ret;
  398. }
  399. Status InnerSession::RunGraphAsync(uint32_t graph_id, const std::vector<ge::Tensor> &inputs,
  400. RunAsyncCallback callback) {
  401. UpdateThreadContext(graph_id);
  402. GELOGI("[InnerSession:%lu] run graph on session, graph_id=%u.", session_id_, graph_id);
  403. Status ret = graph_manager_.RunGraphAsync(graph_id, inputs, session_id_, callback);
  404. if (ret != SUCCESS) {
  405. GELOGE(ret, "[Run][GraphAsync]failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  406. REPORT_CALL_ERROR("E19999",
  407. "GraphManager RunGraphAsync failed, InnerSession:%lu graph_id=%u.", session_id_, graph_id);
  408. return ret;
  409. }
  410. GELOGI("[InnerSession:%lu] run graph success, graph_id=%u.", session_id_, graph_id);
  411. return ret;
  412. }
  413. const GraphManager &InnerSession::getGraphManagerObj() const { return graph_manager_; }
  414. void InnerSession::UpdateThreadContext(const std::map<std::string, std::string> &options) {
  415. GetThreadLocalContext().SetGlobalOption(GetMutableGlobalOptions());
  416. GetThreadLocalContext().SetSessionOption(options_);
  417. GetThreadLocalContext().SetGraphOption(options);
  418. GetContext().SetSessionId(session_id_);
  419. SetRtSocVersion();
  420. }
  421. void InnerSession::UpdateThreadContext(uint32_t graph_id) {
  422. auto options = graph_manager_.GetGraphOptions(graph_id);
  423. if (options == nullptr) {
  424. GELOGW("graph level options is null.");
  425. UpdateThreadContext(std::map<std::string, std::string>{});
  426. } else {
  427. UpdateThreadContext(*options);
  428. }
  429. }
  430. bool InnerSession::IsGraphNeedRebuild(uint32_t graph_id) {
  431. UpdateThreadContext(graph_id);
  432. return graph_manager_.IsGraphNeedRebuild(graph_id);
  433. }
  434. Status InnerSession::GetAllVariables(std::map<std::string, GeTensorDesc> &all_variables) {
  435. return VarManager::Instance(session_id_)->GetAllVariables(all_variables);
  436. }
  437. Status InnerSession::GenCheckPointGraph(const std::map<std::string, GeTensorDesc> &all_variables, Graph &graph) {
  438. return graph_manager_.GenCheckPointGraph(all_variables, graph);
  439. }
  440. Status InnerSession::SaveVariables(const Graph &graph, const std::vector<std::string> &var_names,
  441. const std::vector<Tensor> &outputs, std::vector<Tensor> &var_values) {
  442. return graph_manager_.SaveVariables(graph, var_names, outputs, var_values);
  443. }
  444. Status InnerSession::AddDumpProperties(const DumpProperties &dump_properties) {
  445. if (!is_dump_server_inited_) {
  446. if (dump_properties.IsDumpOpen() || dump_properties.IsOpDebugOpen()) {
  447. GE_IF_BOOL_EXEC(AdxDataDumpServerInit() != kDumpStatus,
  448. GELOGE(PARAM_INVALID, "[Init][AdxDataDumpServer] failed, session_id:%lu.", session_id_);
  449. return PARAM_INVALID)
  450. GELOGI("Init adx data dump server success");
  451. is_dump_server_inited_ = true;
  452. }
  453. }
  454. DumpManager::GetInstance().AddDumpProperties(session_id_, dump_properties);
  455. return SUCCESS;
  456. }
  457. Status InnerSession::RemoveDumpProperties() {
  458. DumpManager::GetInstance().RemoveDumpProperties(session_id_);
  459. if (is_dump_server_inited_ && DumpManager::GetInstance().GetDumpPropertiesMap().empty()) {
  460. GE_IF_BOOL_EXEC(AdxDataDumpServerUnInit() != kDumpStatus,
  461. GELOGE(PARAM_INVALID, "[UnInit][AdxDataDumpServer] failed, session_id:%lu.", session_id_);
  462. REPORT_INNER_ERROR("E19999", "RemoveDumpProperties failed because AdxDataDumpServerUnInit failed,"
  463. "session_id:%lu", session_id_);
  464. return PARAM_INVALID)
  465. GELOGI("UnInit adx data dump server success");
  466. is_dump_server_inited_ = false;
  467. }
  468. return SUCCESS;
  469. }
  470. void InnerSession::SetRtSocVersion() {
  471. const auto &global_options = GetMutableGlobalOptions();
  472. auto it = global_options.find(ge::SOC_VERSION);
  473. if (it != global_options.end()) {
  474. const char *soc_version = it->second.c_str();
  475. rtError_t rt_ret = rtSetSocVersion(soc_version);
  476. if (rt_ret != RT_ERROR_NONE) {
  477. GELOGW("Set soc version %s failed. ret:0x%X", soc_version, rt_ret);
  478. }
  479. GELOGI("Set soc version %s success.", soc_version);
  480. }
  481. }
  482. } // namespace ge

图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示