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.

ge_executor.cc 43 kB

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
5 years ago
5 years ago
5 years ago
5 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
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
4 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
4 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
4 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
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 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
5 years ago
4 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
4 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
4 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
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 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
4 years ago
5 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
4 years ago
5 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
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153
  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 "executor/ge_executor.h"
  17. #include <cce/cce.h>
  18. #include <cce/compiler_stub.h>
  19. #include <ctime>
  20. #include <iostream>
  21. #include "common/debug/log.h"
  22. #include "common/ge/ge_util.h"
  23. #include "common/helper/model_helper.h"
  24. #include "common/profiling/profiling_manager.h"
  25. #include "common/dump/dump_manager.h"
  26. #include "common/util.h"
  27. #include "framework/common/debug/ge_log.h"
  28. #include "framework/common/util.h"
  29. #include "graph/execute/graph_execute.h"
  30. #include "graph/load/graph_loader.h"
  31. #include "graph/load/new_model_manager/davinci_model_parser.h"
  32. #include "graph/load/new_model_manager/model_manager.h"
  33. #include "graph/manager/graph_mem_allocator.h"
  34. #include "graph/model.h"
  35. #include "graph/utils/graph_utils.h"
  36. #include "mmpa/mmpa_api.h"
  37. #include "single_op/single_op_manager.h"
  38. #include "graph/manager/graph_var_manager.h"
  39. #include "graph/load/new_model_manager/davinci_model.h"
  40. #include "opskernel_manager/ops_kernel_builder_manager.h"
  41. using std::string;
  42. using std::vector;
  43. namespace {
  44. const size_t kDynamicBatchSizeVecSize = 1;
  45. const size_t kStaticBatchInfoSize = 1;
  46. const size_t kDynamicImageSizeVecSize = 2;
  47. const size_t kDynamicImageSizeInputSize = 2;
  48. const char *const kBatchLabel = "Batch_";
  49. void GetGeTensorDescFromDomiInfo(std::vector<ge::TensorDesc> &ge_descs,
  50. const std::vector<ge::InputOutputDescInfo> &domi_descs,
  51. const std::vector<uint32_t> &formats) {
  52. uint32_t idx = 0;
  53. for (auto desc_item : domi_descs) {
  54. ge::TensorDesc ge_desc;
  55. ge_desc.SetName(desc_item.name);
  56. ge_desc.SetDataType(static_cast<ge::DataType>(desc_item.data_type));
  57. ge_desc.SetFormat(static_cast<ge::Format>(formats[idx]));
  58. std::vector<int64_t> shape_dims;
  59. for (auto dim : desc_item.shape_info.dims) {
  60. shape_dims.push_back(dim);
  61. }
  62. ge::Shape ge_shape(shape_dims);
  63. ge_desc.SetShape(ge_shape);
  64. ge_desc.SetSize(desc_item.size);
  65. ge_descs.emplace_back(ge_desc);
  66. ++idx;
  67. }
  68. }
  69. void GetDomiInputData(const ge::RunModelData &input_data, ge::InputData &inputs) {
  70. inputs.index = input_data.index;
  71. inputs.model_id = input_data.modelId;
  72. inputs.timestamp = input_data.timestamp;
  73. inputs.timeout = input_data.timeout;
  74. inputs.request_id = input_data.request_id;
  75. for (const auto &data_item : input_data.blobs) {
  76. ge::DataBuffer dataBuf{data_item.data, data_item.length, data_item.isDataSupportMemShare};
  77. inputs.blobs.emplace_back(dataBuf);
  78. }
  79. }
  80. void GetDomiOutputData(const ge::RunModelData &output_data, ge::OutputData &outputs) {
  81. outputs.index = output_data.index;
  82. outputs.model_id = output_data.modelId;
  83. for (const auto &data_item : output_data.blobs) {
  84. ge::DataBuffer dataBuf(data_item.data, data_item.length, data_item.isDataSupportMemShare);
  85. outputs.blobs.emplace_back(dataBuf);
  86. }
  87. }
  88. void SetDynamicInputDataFlag(const ge::RunModelData &input_data, const std::vector<std::vector<int64_t>> batch_info,
  89. ge::InputData &inputs) {
  90. inputs.is_dynamic_batch = true;
  91. std::string batch_label;
  92. size_t match_idx = 0;
  93. for (size_t i = 0; i < batch_info.size(); ++i) {
  94. // dynamic_dims
  95. if (input_data.dynamic_dims.size() != 0) {
  96. bool is_match = true;
  97. for (size_t j = 0; j < static_cast<size_t>(input_data.dynamic_dims.size()); ++j) {
  98. if (static_cast<uint64_t>(batch_info[i][j]) != input_data.dynamic_dims[j]) {
  99. is_match = false;
  100. break;
  101. }
  102. }
  103. if (is_match) {
  104. match_idx = i;
  105. break;
  106. }
  107. // dynamic_batch_size
  108. } else if (batch_info[i].size() == kDynamicBatchSizeVecSize &&
  109. batch_info[i][0] == static_cast<int64_t>(input_data.dynamic_batch_size)) {
  110. match_idx = i;
  111. break;
  112. // dynamic_image_size
  113. } else if (batch_info[i].size() == kDynamicImageSizeVecSize &&
  114. batch_info[i][0] == static_cast<int64_t>(input_data.dynamic_image_height) &&
  115. batch_info[i][1] == static_cast<int64_t>(input_data.dynamic_image_width)) {
  116. match_idx = i;
  117. break;
  118. }
  119. }
  120. batch_label = kBatchLabel + std::to_string(match_idx);
  121. inputs.batch_label = batch_label;
  122. GELOGI("current batch label:%s", batch_label.c_str());
  123. }
  124. bool IsDynamicBatchSizeMatchModel(uint64_t batch_size, const vector<std::vector<int64_t>> &batch_info) {
  125. if (batch_info.empty()) {
  126. GELOGE(ge::FAILED, "Dynamic batch info is empty.");
  127. return false;
  128. }
  129. for (auto batch : batch_info) {
  130. if (batch.size() != kDynamicBatchSizeVecSize) {
  131. GELOGE(ge::FAILED, "Dynamic batch param num is %zu, current batch size is %zu.", kDynamicBatchSizeVecSize,
  132. batch.size());
  133. return false;
  134. }
  135. if (batch[0] == static_cast<int64_t>(batch_size)) {
  136. return true;
  137. }
  138. }
  139. GELOGE(ge::FAILED, "Dynamic batch %lu can not match the gear of model.", batch_size);
  140. return false;
  141. }
  142. bool IsDynamicImageSizeMatchModel(uint64_t image_height, uint64_t image_width,
  143. const vector<std::vector<int64_t>> &batch_info) {
  144. if (batch_info.empty()) {
  145. GELOGE(ge::FAILED, "Dynamic batch info is empty.");
  146. return false;
  147. }
  148. for (auto resolution : batch_info) {
  149. if (resolution.size() != kDynamicImageSizeVecSize) {
  150. GELOGE(ge::FAILED, "Dynamic resolution param num is %zu, current resolution size is %zu.",
  151. kDynamicImageSizeVecSize, resolution.size());
  152. return false;
  153. }
  154. if (resolution[0] == static_cast<int64_t>(image_height) && resolution[1] == static_cast<int64_t>(image_width)) {
  155. return true;
  156. }
  157. }
  158. GELOGE(ge::FAILED, "Dynamic resolution (%lu,%lu) can not match the gear of model.", image_height, image_width);
  159. return false;
  160. }
  161. bool IsDynmaicDimsSizeMatchModel(const vector<uint64_t> cur_dynamic_dims,
  162. const vector<vector<int64_t>> &batch_info) {
  163. if (batch_info.empty()) {
  164. GELOGE(ge::FAILED, "Dynamic batch info is empty.");
  165. return false;
  166. }
  167. bool find_match = false;
  168. for (auto resolution : batch_info) {
  169. if (cur_dynamic_dims.size() != resolution.size()) {
  170. GELOGE(ge::FAILED, "Cur dynamic dims param num is %zu, current resolution size is %zu.",
  171. cur_dynamic_dims.size(), resolution.size());
  172. return false;
  173. }
  174. bool flag = true;
  175. for (std::size_t i = 0; i < resolution.size(); ++i) {
  176. if (cur_dynamic_dims[i] != static_cast<uint64_t>(resolution[i])) {
  177. flag = false;
  178. break;
  179. }
  180. }
  181. if (flag) {
  182. find_match = true;
  183. break;
  184. }
  185. }
  186. if (!find_match) {
  187. GELOGE(ge::FAILED, "choose dynamic dims can not match the gear of model.");
  188. }
  189. return find_match;
  190. }
  191. } // namespace
  192. namespace ge {
  193. bool GeExecutor::isInit_ = false;
  194. class ModelListenerAdapter : public ModelListener {
  195. public:
  196. domi::Status OnComputeDone(uint32_t model_id, uint32_t dataIndex, uint32_t resultCode,
  197. std::vector<ge::OutputTensorInfo> &outputs) {
  198. if (listener == nullptr) {
  199. GELOGE(ge::FAILED, "listener is null.");
  200. return FAILED;
  201. }
  202. return listener->OnComputeDone(model_id, dataIndex, resultCode, outputs);
  203. }
  204. std::shared_ptr<ge::ModelListener> listener;
  205. };
  206. GeExecutor::GeExecutor() {}
  207. Status GeExecutor::Initialize() {
  208. GELOGI("Init GeExecutor begin.");
  209. if (isInit_) {
  210. GELOGW("Already initialized, no need to be initialized again.");
  211. return ge::SUCCESS;
  212. }
  213. std::vector<rtMemType_t> mem_type(1, RT_MEMORY_HBM);
  214. mem_type.push_back(RT_MEMORY_P2P_DDR);
  215. auto ret = MemManager::Instance().Initialize(mem_type);
  216. if (ret != SUCCESS) {
  217. GELOGE(ret, "Memory Manager init failed.");
  218. return ret;
  219. }
  220. GE_CHK_STATUS_RET(OpsKernelBuilderManager::Instance().Initialize({}, false),
  221. "Failed to initialize OpsKernelBuilders");
  222. // Start profiling
  223. Options profiling_options;
  224. profiling_options.device_id = 0;
  225. profiling_options.job_id = "";
  226. ProfilingManager::Instance().Init(profiling_options);
  227. isInit_ = true;
  228. GELOGI("Init GeExecutor over.");
  229. return ge::SUCCESS;
  230. }
  231. Status GeExecutor::Finalize() {
  232. GELOGI("Uninit GeExecutor begin.");
  233. if (isInit_ == false) {
  234. GELOGW("GeExecutor has not been initialized.");
  235. return ge::SUCCESS;
  236. }
  237. (void) OpsKernelBuilderManager::Instance().Finalize();
  238. // Stop profiling
  239. if (ProfilingManager::Instance().ProfilingOn()) {
  240. ProfilingManager::Instance().StopProfiling();
  241. ProfilingManager::Instance().PluginUnInit(GE_PROFILING_MODULE);
  242. }
  243. GELOGI("Uninit GeExecutor over.");
  244. return ge::SUCCESS;
  245. }
  246. Status GeExecutor::SetDynamicBatchSize(uint32_t model_id, void *dynamic_input_addr, uint64_t length,
  247. uint64_t batch_size) {
  248. if (dynamic_input_addr == nullptr) {
  249. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID, "Dynamic input addr is nullptr!");
  250. return ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID;
  251. }
  252. uint64_t size = sizeof(uint32_t);
  253. if (length < size) {
  254. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID, "Dynamic input size [%lu] is less than [%lu]!", length, size);
  255. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  256. }
  257. if (length >= sizeof(uint64_t)) {
  258. size = sizeof(uint64_t);
  259. }
  260. // Verify whether the input dynamic batch matches the model gear
  261. std::vector<std::vector<int64_t>> batch_info;
  262. std::vector<uint64_t> batch_num{batch_size};
  263. int32_t dynamic_type = static_cast<int32_t>(FIXED);
  264. Status ret = GraphExecutor::GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  265. if (ret != SUCCESS) {
  266. GELOGE(ret, "Get dynamic input info failed.");
  267. return ret;
  268. }
  269. if (!IsDynamicBatchSizeMatchModel(batch_size, batch_info)) {
  270. GELOGE(ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID, "The current dynamic input does not match the gear of the model.");
  271. return ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID;
  272. }
  273. ret = GraphExecutor::SetDynamicSize(model_id, batch_num, static_cast<int32_t>(DYNAMIC_BATCH));
  274. if (ret != SUCCESS) {
  275. GELOGE(ret, "Set dynamic size failed");
  276. return ret;
  277. }
  278. // memcpy dynamic_batch_size from host to device
  279. rtError_t rt_ret = rtMemcpy(dynamic_input_addr, length, &batch_size, size, RT_MEMCPY_HOST_TO_DEVICE);
  280. if (rt_ret != RT_ERROR_NONE) {
  281. GELOGE(rt_ret, "memcpy dynamic batch input data failed! ret: 0x%X", rt_ret);
  282. return RT_ERROR_TO_GE_STATUS(rt_ret);
  283. }
  284. return SUCCESS;
  285. }
  286. Status GeExecutor::SetDynamicImageSize(uint32_t model_id, void *dynamic_input_addr, uint64_t length,
  287. uint64_t image_height, uint64_t image_width) {
  288. if (dynamic_input_addr == nullptr) {
  289. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID, "Dynamic input addr is nullptr!");
  290. return ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID;
  291. }
  292. uint64_t dynamic_input_size = kDynamicImageSizeInputSize * sizeof(uint32_t);
  293. if (length < dynamic_input_size) {
  294. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID,
  295. "Dynamic input size [%lu] is less than [%lu]!", length, dynamic_input_size);
  296. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  297. }
  298. uint64_t size = sizeof(uint32_t);
  299. if (length >= kDynamicImageSizeInputSize * sizeof(uint64_t)) {
  300. size = sizeof(uint64_t);
  301. }
  302. // Verify whether the input dynamic resolution matches the model gear
  303. std::vector<std::vector<int64_t>> batch_info;
  304. std::vector<uint64_t> batch_num{image_height, image_width};
  305. int32_t dynamic_type = static_cast<int32_t>(FIXED);
  306. Status ret = GraphExecutor::GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  307. if (ret != SUCCESS) {
  308. GELOGE(ret, "Get dynamic input info failed.");
  309. return ret;
  310. }
  311. if (!IsDynamicImageSizeMatchModel(image_height, image_width, batch_info)) {
  312. GELOGE(ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID, "The current dynamic input does not match the gear of the model.");
  313. return ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID;
  314. }
  315. ret = GraphExecutor::SetDynamicSize(model_id, batch_num, static_cast<int32_t>(DYNAMIC_IMAGE));
  316. if (ret != SUCCESS) {
  317. GELOGE(ret, "Set dynamic size failed");
  318. return ret;
  319. }
  320. // Memcpy dynamic resolution height from host to device
  321. rtError_t rt_ret =
  322. rtMemcpy(dynamic_input_addr, size, &image_height, size, RT_MEMCPY_HOST_TO_DEVICE);
  323. if (rt_ret != RT_ERROR_NONE) {
  324. GELOGE(rt_ret, "memcpy dynamic resolution input data failed! ret: 0x%X", rt_ret);
  325. return RT_ERROR_TO_GE_STATUS(rt_ret);
  326. }
  327. uint64_t remain_size = length - size;
  328. // Memcpy dynamic resolution width from host to device
  329. rt_ret = rtMemcpy(reinterpret_cast<void *>(reinterpret_cast<uint8_t *>(dynamic_input_addr) + size),
  330. remain_size, &image_width, size, RT_MEMCPY_HOST_TO_DEVICE);
  331. if (rt_ret != RT_ERROR_NONE) {
  332. GELOGE(rt_ret, "memcpy dynamic resolution input data failed!");
  333. return RT_ERROR_TO_GE_STATUS(rt_ret);
  334. }
  335. return SUCCESS;
  336. }
  337. Status GeExecutor::SetDynamicDims(uint32_t model_id, void *dynamic_input_addr, uint64_t length,
  338. const vector<uint64_t> &dynamic_dims) {
  339. if (dynamic_input_addr == nullptr) {
  340. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID, "Dynamic input addr is nullptr!");
  341. return ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID;
  342. }
  343. vector<uint64_t> cur_dynamic_dims;
  344. Status ret = GetCurDynamicDims(model_id, dynamic_dims, cur_dynamic_dims);
  345. if (ret != SUCCESS) {
  346. GELOGE(ret, "Set cur gear dynamic dims failed");
  347. return ret;
  348. }
  349. std::vector<std::vector<int64_t>> batch_info;
  350. int32_t dynamic_type = static_cast<int32_t>(FIXED);
  351. ret = GraphExecutor::GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  352. if (ret != SUCCESS) {
  353. GELOGE(ret, "Get dynamic input info failed.");
  354. return ret;
  355. }
  356. if (!IsDynmaicDimsSizeMatchModel(cur_dynamic_dims, batch_info)) {
  357. GELOGE(ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID, "The current dynamic input does not match the gear of the model.");
  358. return ACL_ERROR_GE_DYNAMIC_BATCH_SIZE_INVALID;
  359. }
  360. ret = GraphExecutor::SetDynamicSize(model_id, cur_dynamic_dims, static_cast<int32_t>(DYNAMIC_DIMS));
  361. if (ret != SUCCESS) {
  362. GELOGE(ret, "Set dynamic size failed");
  363. return ret;
  364. }
  365. size_t dynamic_dim_num = cur_dynamic_dims.size();
  366. uint64_t dynamic_input_size = static_cast<uint64_t>(dynamic_dim_num * sizeof(uint32_t));
  367. if (length < dynamic_input_size) {
  368. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID,
  369. "Dynamic input size [%lu] is less than [%lu]!", length, dynamic_input_size);
  370. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  371. }
  372. uint64_t size = sizeof(uint32_t);
  373. if (length >= dynamic_dim_num * sizeof(uint64_t)) {
  374. size = sizeof(uint64_t);
  375. }
  376. rtError_t rt_ret;
  377. for (uint32_t i = 0; i < dynamic_dim_num; ++i) {
  378. // Memcpy dynamic dim[i] from host to device
  379. rt_ret = rtMemcpy(reinterpret_cast<void *>(reinterpret_cast<uint8_t *>(dynamic_input_addr) + size * i),
  380. length - size * i, &cur_dynamic_dims[i], size, RT_MEMCPY_HOST_TO_DEVICE);
  381. if (rt_ret != RT_ERROR_NONE) {
  382. GELOGE(rt_ret, "memcpy dynamic resolution input data failed!");
  383. return RT_ERROR_TO_GE_STATUS(rt_ret);
  384. }
  385. }
  386. return SUCCESS;
  387. }
  388. Status GeExecutor::GetCurDynamicDims(uint32_t model_id, const vector<uint64_t> &dynamic_dims,
  389. vector<uint64_t> &cur_dynamic_dims) {
  390. cur_dynamic_dims.clear();
  391. vector<ge::TensorDesc> input_desc;
  392. vector<ge::TensorDesc> output_desc;
  393. auto ret = GetModelDescInfo(model_id, input_desc, output_desc);
  394. if (ret != ge::SUCCESS) {
  395. GELOGE(ret, "GetModelDescInfo failed.");
  396. return ret;
  397. }
  398. vector<string> user_designate_shape_order;
  399. vector<int64_t> all_data_dims;
  400. ret = GetUserDesignateShapeOrder(model_id, user_designate_shape_order);
  401. if (ret != ge::SUCCESS) {
  402. GELOGE(ret, "GetUserDesignateShapeOrder failed.");
  403. return ret;
  404. }
  405. for (auto &data_name : user_designate_shape_order) {
  406. for (auto &desc : input_desc) {
  407. if (desc.GetName() == data_name) {
  408. for (auto dim : desc.GetShape().GetDims()) {
  409. all_data_dims.push_back(dim);
  410. }
  411. break;
  412. }
  413. }
  414. }
  415. if (dynamic_dims.size() != all_data_dims.size()){
  416. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID,
  417. "Dynamic input size [%lu] is not equal with all data dims size [%lu]!",
  418. dynamic_dims.size(), all_data_dims.size());
  419. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  420. }
  421. for (std::size_t i = 0; i < all_data_dims.size(); ++i) {
  422. if (all_data_dims[i] < 0) {
  423. cur_dynamic_dims.push_back(dynamic_dims[i]);
  424. } else if (static_cast<uint64_t>(all_data_dims[i]) != dynamic_dims[i]) {
  425. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID, "Static dims should be same, index: %zu value: %d should be %d",
  426. i, dynamic_dims[i], all_data_dims[i]);
  427. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  428. }
  429. }
  430. return SUCCESS;
  431. }
  432. Status GeExecutor::GetCurShape(const uint32_t model_id, std::vector<int64_t> &batch_info, int32_t &dynamic_type) {
  433. GELOGI("Begin to get current shape");
  434. if (!isInit_) {
  435. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  436. return ACL_ERROR_GE_EXEC_NOT_INIT;
  437. }
  438. Status ret = GraphExecutor::GetCurShape(model_id, batch_info, dynamic_type);
  439. if (ret != SUCCESS) {
  440. GELOGE(ret, "Get current shape failed");
  441. return ret;
  442. }
  443. return SUCCESS;
  444. }
  445. Status GeExecutor::SetDynamicAippData(uint32_t model_id, void *dynamic_input_addr, uint64_t length,
  446. const std::vector<kAippDynamicBatchPara> &aippBatchPara,
  447. const kAippDynamicPara &aippParms) {
  448. GELOGI("Enter to SetDynamicAippData.");
  449. if (dynamic_input_addr == nullptr) {
  450. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID, "Dynamic aipp input addr is nullptr!");
  451. return ACL_ERROR_GE_DYNAMIC_INPUT_ADDR_INVALID;
  452. }
  453. if (aippBatchPara.empty()) {
  454. GELOGE(ACL_ERROR_GE_AIPP_BATCH_EMPTY, "aippBatchPara is empty.");
  455. return ACL_ERROR_GE_AIPP_BATCH_EMPTY;
  456. }
  457. uint64_t batch_num = aippBatchPara.size();
  458. uint64_t real_aippParms_size = sizeof(kAippDynamicPara) - sizeof(kAippDynamicBatchPara);
  459. uint64_t struct_len = batch_num * sizeof(kAippDynamicBatchPara) + real_aippParms_size;
  460. GELOGI(
  461. "Get acl input dynamic aipp data, model_id is %u, length is %lu,"
  462. "batch num is %lu, struct_len is %lu",
  463. model_id, length, batch_num, struct_len);
  464. if (struct_len > length) {
  465. GELOGE(ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID,
  466. "input dynamic aipp param len [%lu] is larger than aipp_data size [%lu]", struct_len, length);
  467. return ACL_ERROR_GE_DYNAMIC_INPUT_LENGTH_INVALID;
  468. }
  469. // Memcpy real kAippDynamicBatchPara from host to device
  470. rtError_t rt_ret = rtMemcpy(dynamic_input_addr, length, &aippParms, real_aippParms_size, RT_MEMCPY_HOST_TO_DEVICE);
  471. if (rt_ret != RT_ERROR_NONE) {
  472. GELOGE(rt_ret, "memcpy real_aippParms_size failed! ret: 0x%X", rt_ret);
  473. return RT_ERROR_TO_GE_STATUS(rt_ret);
  474. }
  475. uint64_t remain_len = length - real_aippParms_size;
  476. uint8_t *aipp_batch_para_dev = reinterpret_cast<uint8_t *>(dynamic_input_addr) + real_aippParms_size;
  477. for (uint64_t i = 0; i < batch_num; ++i) {
  478. rt_ret = rtMemcpy(reinterpret_cast<void *>(aipp_batch_para_dev + i * sizeof(kAippDynamicBatchPara)),
  479. (remain_len - i * sizeof(kAippDynamicBatchPara)), &(aippBatchPara[i]),
  480. sizeof(kAippDynamicBatchPara), RT_MEMCPY_HOST_TO_DEVICE);
  481. if (rt_ret != RT_ERROR_NONE) {
  482. GELOGE(rt_ret, "memcpy kAippDynamicBatchPara input data failed! ret: 0x%X", rt_ret);
  483. return RT_ERROR_TO_GE_STATUS(rt_ret);
  484. }
  485. }
  486. return SUCCESS;
  487. }
  488. // Load model
  489. Status GeExecutor::LoadModelOffline(uint32_t &model_id, const std::string &path, const std::string &key,
  490. int32_t priority, std::shared_ptr<ge::ModelListener> listener) {
  491. GELOGI("load model offline begin.");
  492. if (!isInit_) {
  493. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  494. return ACL_ERROR_GE_EXEC_NOT_INIT;
  495. }
  496. string filePath = RealPath(path.c_str());
  497. if (filePath.empty()) {
  498. GELOGE(ACL_ERROR_GE_EXEC_MODEL_PATH_INVALID,
  499. "File path is invalid. please check your text file '%s'.", path.c_str());
  500. return ACL_ERROR_GE_EXEC_MODEL_PATH_INVALID;
  501. }
  502. std::shared_ptr<ModelListenerAdapter> listener_adapter = MakeShared<ModelListenerAdapter>();
  503. if (listener_adapter == nullptr) {
  504. GELOGE(ACL_ERROR_GE_MEMORY_ALLOCATION, "ModelListenerAdapter make shared failed!");
  505. return ACL_ERROR_GE_MEMORY_ALLOCATION;
  506. }
  507. listener_adapter->listener = listener;
  508. Status ret = GraphLoader::LoadModelFromFile(path, key, priority, listener_adapter, model_id);
  509. if (ret != SUCCESS) {
  510. GELOGE(ret, "[GeExecutor] LoadModelFromFile failed");
  511. return ACL_ERROR_GE_LOAD_MODEL;
  512. }
  513. return SUCCESS;
  514. }
  515. Status GeExecutor::LoadModel(uint32_t &model_id, const ModelData &model_data,
  516. std::shared_ptr<ge::ModelListener> listener) {
  517. GELOGI("Load model begin.");
  518. if (!isInit_) {
  519. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  520. return ACL_ERROR_GE_EXEC_NOT_INIT;
  521. }
  522. std::shared_ptr<ModelListenerAdapter> listener_adapter = MakeShared<ModelListenerAdapter>();
  523. if (listener_adapter == nullptr) {
  524. GELOGE(ACL_ERROR_GE_MEMORY_ALLOCATION, "ModelListenerAdapter make shared failed!");
  525. return ACL_ERROR_GE_MEMORY_ALLOCATION;
  526. }
  527. listener_adapter->listener = listener;
  528. Status ret = GraphLoader::LoadModel(model_data, listener_adapter, model_id);
  529. if (ret != SUCCESS) {
  530. GELOGE(ret, "[GeExecutor] LoadModel failed.");
  531. return ACL_ERROR_GE_LOAD_MODEL;
  532. }
  533. return ret;
  534. }
  535. Status GeExecutor::UnloadModel(uint32_t model_id) {
  536. GELOGI("unload model %u begin.", model_id);
  537. if (!isInit_) {
  538. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  539. return ACL_ERROR_GE_EXEC_NOT_INIT;
  540. }
  541. Status ret = GraphLoader::DestroyAicpuSessionForInfer(model_id);
  542. if (ret != SUCCESS) {
  543. GELOGE(ret, "[GraphLoader] DestroyAicpuSessionForInfer failed. model id: %u", model_id);
  544. return ACL_ERROR_GE_INTERNAL_ERROR;
  545. }
  546. std::shared_ptr<DavinciModel> davinci_model = ModelManager::GetInstance()->GetModel(model_id);
  547. if (davinci_model != nullptr) {
  548. uint64_t session_id = davinci_model->GetSessionId();
  549. VarManagerPool::Instance().RemoveVarManager(session_id);
  550. }
  551. ret = GraphLoader::UnloadModel(model_id);
  552. if (ret != SUCCESS) {
  553. GELOGE(ret, "[GraphLoader] DestroyAicpuSessionForInfer failed. model id: %u", model_id);
  554. return ACL_ERROR_GE_UNLOAD_MODEL;
  555. }
  556. return SUCCESS;
  557. }
  558. Status GeExecutor::RunModel(const ge::RunModelData &input_data, ge::RunModelData &output_data) {
  559. GELOGI("run model begin.");
  560. if (!isInit_) {
  561. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  562. return ACL_ERROR_GE_EXEC_NOT_INIT;
  563. }
  564. InputData inputs;
  565. GetDomiInputData(input_data, inputs);
  566. OutputData outputs;
  567. GetDomiOutputData(output_data, outputs);
  568. return GraphExecutor::DataInput(inputs, outputs);
  569. }
  570. // Get input and output descriptor
  571. Status GeExecutor::GetModelDescInfo(uint32_t model_id, std::vector<ge::TensorDesc> &input_desc,
  572. std::vector<ge::TensorDesc> &output_desc, bool new_model_desc) {
  573. GELOGI("get model desc info begin.");
  574. if (!isInit_) {
  575. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  576. return ACL_ERROR_GE_EXEC_NOT_INIT;
  577. }
  578. std::vector<InputOutputDescInfo> input_desc_infos;
  579. std::vector<InputOutputDescInfo> output_desc_infos;
  580. std::vector<uint32_t> input_formats;
  581. std::vector<uint32_t> output_formats;
  582. Status ret = GraphExecutor::GetInputOutputDescInfo(model_id, input_desc_infos, output_desc_infos, input_formats,
  583. output_formats, new_model_desc);
  584. if (ret != domi::SUCCESS) {
  585. GELOGE(ret, "GetInputOutputDescInfo failed. ret = %u", ret);
  586. return ACL_ERROR_GE_GET_TENSOR_INFO;
  587. }
  588. if (input_formats.size() != input_desc_infos.size()) {
  589. GELOGE(ACL_ERROR_GE_PARAM_INVALID,
  590. "input_formats size %zu is not equal to input_desc_infos size %zu.",
  591. input_formats.size(), input_desc_infos.size());
  592. return ACL_ERROR_GE_PARAM_INVALID;
  593. }
  594. if (output_formats.size() != output_desc_infos.size()) {
  595. GELOGE(ACL_ERROR_GE_PARAM_INVALID, "output_formats size %zu is not equal to output_desc_infos size %zu.",
  596. output_formats.size(), output_desc_infos.size());
  597. return ACL_ERROR_GE_PARAM_INVALID;
  598. }
  599. // Transfer data to TensorDesc
  600. GetGeTensorDescFromDomiInfo(input_desc, input_desc_infos, input_formats);
  601. GetGeTensorDescFromDomiInfo(output_desc, output_desc_infos, output_formats);
  602. GELOGI("get model desc info end.");
  603. return ge::SUCCESS;
  604. }
  605. ///
  606. /// @ingroup ge
  607. /// @brief Get dynamic batch_info
  608. /// @param [in] model_id
  609. /// @param [out] batch_info
  610. /// @param [out] dynamic_type
  611. /// @return execute result
  612. ///
  613. Status GeExecutor::GetDynamicBatchInfo(uint32_t model_id, std::vector<std::vector<int64_t>> &batch_info,
  614. int32_t &dynamic_type) {
  615. GELOGI("Begin to get dynamic batch info.");
  616. if (!isInit_) {
  617. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  618. return ACL_ERROR_GE_EXEC_NOT_INIT;
  619. }
  620. Status ret = GraphExecutor::GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  621. if (ret != SUCCESS) {
  622. GELOGE(ret, "GetDynamicBatchInfo failed.");
  623. return ret;
  624. }
  625. GELOGI("Get dynamic batch info succ.");
  626. return SUCCESS;
  627. }
  628. ///
  629. /// @ingroup ge
  630. /// @brief Get combined dynamic dims info
  631. /// @param [in] model_id
  632. /// @param [out] batch_info
  633. /// @return execute result
  634. ///
  635. Status GeExecutor::GetCombinedDynamicDims(uint32_t model_id, vector<vector<int64_t>> &batch_info) {
  636. GELOGI("Begin to get combined dynamic dims info.");
  637. if (!isInit_) {
  638. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  639. return ACL_ERROR_GE_EXEC_NOT_INIT;
  640. }
  641. Status ret = GraphExecutor::GetCombinedDynamicDims(model_id, batch_info);
  642. if (ret != SUCCESS) {
  643. GELOGE(ret, "GetCombinedDynamicDims failed.");
  644. return ret;
  645. }
  646. GELOGI("Get combined dynamic dims succ.");
  647. return SUCCESS;
  648. }
  649. ///
  650. /// @ingroup ge
  651. /// @brief Get user designeate shape order
  652. /// @param [in] model_id
  653. /// @param [out] user_designate_shape_order
  654. /// @return execute result
  655. ///
  656. Status GeExecutor::GetUserDesignateShapeOrder(uint32_t model_id, vector<string> &user_designate_shape_order) {
  657. GELOGI("Begin to get user designate shape info.");
  658. if (!isInit_) {
  659. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  660. return ACL_ERROR_GE_EXEC_NOT_INIT;
  661. }
  662. Status ret = GraphExecutor::GetUserDesignateShapeOrder(model_id, user_designate_shape_order);
  663. if (ret != SUCCESS) {
  664. GELOGE(ret, "GetUserDesignateShapeOrder failed.");
  665. return ret;
  666. }
  667. GELOGI("Get user designate shape order succ.");
  668. return SUCCESS;
  669. }
  670. ///
  671. /// @ingroup ge
  672. /// @brief Get AIPP input format
  673. /// @param [in] model_id
  674. /// @param [in] index
  675. /// @param [out] input_format
  676. /// @return execute result
  677. ///
  678. Status GeExecutor::GetAIPPInfo(uint32_t model_id, uint32_t index, AippConfigInfo &aipp_info) {
  679. GELOGI("Begin to GetAIPPInfo.");
  680. if (!isInit_) {
  681. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  682. return ACL_ERROR_GE_EXEC_NOT_INIT;
  683. }
  684. Status ret = GraphExecutor::GetAIPPInfo(model_id, index, aipp_info);
  685. if (ret != SUCCESS) {
  686. GELOGW("GetAIPPInfo is not success.");
  687. return ret;
  688. }
  689. GELOGI("GetAIPPInfo succ.");
  690. return SUCCESS;
  691. }
  692. Status GeExecutor::GetAippType(uint32_t model_id, uint32_t index, InputAippType &type, size_t &aipp_index) {
  693. GELOGI("Begin to get aipp type.");
  694. if (!isInit_) {
  695. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  696. return ACL_ERROR_GE_EXEC_NOT_INIT;
  697. }
  698. Status ret = GraphExecutor::GetAippType(model_id, index, type, aipp_index);
  699. if (ret != SUCCESS) {
  700. GELOGW("Get aipp type is not success.");
  701. return ret;
  702. }
  703. GELOGI("Get aipp type success.");
  704. return SUCCESS;
  705. }
  706. Status GeExecutor::GetModelAttr(uint32_t model_id, std::vector<std::string> &dynamic_output_shape_info) {
  707. GELOGI("Begin to get dynamic batch output shape info");
  708. if (!isInit_) {
  709. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  710. return ACL_ERROR_GE_EXEC_NOT_INIT;
  711. }
  712. Status ret = GraphExecutor::GetModelAttr(model_id, dynamic_output_shape_info);
  713. if (ret != SUCCESS) {
  714. GELOGE(ret, "Get dynamic batch output shape info failed.");
  715. return ret;
  716. }
  717. GELOGI("Get dynamic batch output shape info succ.");
  718. return SUCCESS;
  719. }
  720. Status GeExecutor::GetModelDescInfoForZeroCopy(uint32_t model_id, std::vector<ge::TensorDesc> &input_desc,
  721. std::vector<TensorDesc> &output_desc) {
  722. GELOGI("get model desc info for zero copy begin.");
  723. if (!isInit_) {
  724. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  725. return ACL_ERROR_GE_EXEC_NOT_INIT;
  726. }
  727. std::vector<InputOutputDescInfo> input_desc_infos;
  728. std::vector<InputOutputDescInfo> output_desc_infos;
  729. std::vector<uint32_t> input_formats;
  730. std::vector<uint32_t> output_formats;
  731. Status ret = GraphExecutor::GetInputOutputDescInfoForZeroCopy(model_id, input_desc_infos, output_desc_infos,
  732. input_formats, output_formats);
  733. if (ret != domi::SUCCESS) {
  734. GELOGE(ret, "Get DescInfo from zero copy failed. ret = %u", ret);
  735. return ACL_ERROR_GE_GET_TENSOR_INFO;
  736. }
  737. if (input_formats.size() != input_desc_infos.size()) {
  738. GELOGE(ACL_ERROR_GE_PARAM_INVALID, "input_formats.size() != input_desc_infos.size().");
  739. return ACL_ERROR_GE_PARAM_INVALID;
  740. }
  741. if (output_formats.size() != output_desc_infos.size()) {
  742. GELOGE(ACL_ERROR_GE_PARAM_INVALID, "output_formats.size() != output_desc_infos.size().");
  743. return ACL_ERROR_GE_PARAM_INVALID;
  744. }
  745. GetGeTensorDescFromDomiInfo(input_desc, input_desc_infos, input_formats);
  746. GetGeTensorDescFromDomiInfo(output_desc, output_desc_infos, output_formats);
  747. GELOGI("get model desc info from zero copy end.");
  748. return ge::SUCCESS;
  749. }
  750. Status GeExecutor::CommandHandle(const Command &command) {
  751. GELOGI("command handle begin.");
  752. Status ret = GraphLoader::CommandHandle(command);
  753. if (ret != SUCCESS) {
  754. GELOGE(ACL_ERROR_GE_COMMAND_HANDLE, "CommandHandle: Command Handle failed.");
  755. return ACL_ERROR_GE_COMMAND_HANDLE;
  756. }
  757. return SUCCESS;
  758. }
  759. Status GeExecutor::GetMaxUsedMemory(uint32_t model_id, uint32_t &max_size) {
  760. GELOGI("Get max used memory begin.");
  761. if (!isInit_) {
  762. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  763. return ACL_ERROR_GE_EXEC_NOT_INIT;
  764. }
  765. uint64_t max_mem_size = 0;
  766. Status ret = GraphLoader::GetMaxUsedMemory(model_id, max_mem_size);
  767. max_size = static_cast<uint32_t>(max_mem_size);
  768. return ret;
  769. }
  770. /**
  771. * @ingroup ge
  772. * @brief Load data from model file to memory
  773. * @param [in] const std::string &path: Offline model file path
  774. * @param [out] domi::ModelData &model_data: Offline model memory data
  775. * @return SUCCESS handle successfully / others handle failed
  776. */
  777. Status GeExecutor::LoadDataFromFile(const std::string &path, ModelData &model_data) {
  778. GELOGI("Load data from file begin.");
  779. if (!isInit_) {
  780. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  781. return ACL_ERROR_GE_EXEC_NOT_INIT;
  782. }
  783. string filePath = RealPath(path.c_str());
  784. if (filePath.empty()) {
  785. GELOGE(ACL_ERROR_GE_EXEC_MODEL_PATH_INVALID,
  786. "File path is invalid. please check your text file '%s'.", path.c_str());
  787. return ACL_ERROR_GE_EXEC_MODEL_PATH_INVALID;
  788. }
  789. GELOGI("load modelData from file: %s.", path.c_str());
  790. std::string key_path;
  791. int32_t priority = 0;
  792. Status ret = GraphLoader::LoadDataFromFile(path, key_path, priority, model_data);
  793. if (ret != SUCCESS) {
  794. if (model_data.model_data != nullptr) {
  795. delete[] static_cast<char *>(model_data.model_data);
  796. model_data.model_data = nullptr;
  797. }
  798. }
  799. return ret;
  800. }
  801. /**
  802. * @ingroup ge
  803. * @brief Load model from offline model memory data
  804. * @param [in] domi::ModelData &model_data: Offline model data
  805. void *dev_ptr: Input/Output memory start address
  806. size_t memsize: Input/Output memory length
  807. void *weight_ptr: Weight memory start address
  808. size_t weightsize: Weight memory length
  809. * @param [out] uint32_t &model_id: identification after model loading
  810. * @return SUCCESS handle successfully / others handle failed
  811. */
  812. Status GeExecutor::LoadModelFromData(uint32_t &model_id, const ModelData &model_data, void *dev_ptr, size_t mem_size,
  813. void *weight_ptr, size_t weight_size) {
  814. GELOGI("Load model from data begin.");
  815. if (!isInit_) {
  816. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  817. return ACL_ERROR_GE_EXEC_NOT_INIT;
  818. }
  819. return GraphLoader::LoadModelFromData(model_id, model_data, dev_ptr, mem_size, weight_ptr, weight_size);
  820. }
  821. /**
  822. * @ingroup ge
  823. * @brief Load task list from ModelData with queue.
  824. * @param [out] model_id: model id allocate from manager.
  825. * @param [in] ge_model_data: Model data load from offline model.
  826. * @param [in] input_queue_ids: input queue ids create from user.
  827. * @param [in] output_queue_ids: input queue ids create from user.
  828. * @return: 0 for success / others for fail
  829. */
  830. Status GeExecutor::LoadModelWithQ(uint32_t &model_id, const ModelData &model_data,
  831. const std::vector<uint32_t> &input_queue_ids,
  832. const std::vector<uint32_t> &output_queue_ids) {
  833. GELOGI("Load model with queue begin.");
  834. if (!isInit_) {
  835. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  836. return ACL_ERROR_GE_EXEC_NOT_INIT;
  837. }
  838. return GraphLoader::LoadModelWithQ(model_id, model_data, input_queue_ids, output_queue_ids);
  839. }
  840. /**
  841. * @ingroup ge
  842. * @brief Synchronous execution of offline model(Do not create thread)
  843. * @param [in] uint32_t model_id: Model ID to execute
  844. void* stream: stream to execute
  845. const domi::InputData *input_data: Model input data
  846. bool async_mode: is asynchronize mode.
  847. * @param [out] domi::OutputData *output_data: Model output data
  848. * @return SUCCESS handle successfully / others handle failed
  849. */
  850. Status GeExecutor::ExecModel(uint32_t model_id, void *stream, const ge::RunModelData &run_input_data,
  851. ge::RunModelData &run_output_data, bool async_mode) {
  852. GELOGI("Execute model begin.");
  853. if (!isInit_) {
  854. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  855. return ACL_ERROR_GE_EXEC_NOT_INIT;
  856. }
  857. InputData input_data;
  858. OutputData output_data;
  859. GetDomiInputData(run_input_data, input_data);
  860. GetDomiOutputData(run_output_data, output_data);
  861. if ((run_input_data.dynamic_batch_size != 0) || (run_input_data.dynamic_image_width != 0) ||
  862. (run_input_data.dynamic_image_height != 0) || (run_input_data.dynamic_dims.size() != 0)) {
  863. std::vector<std::vector<int64_t>> batch_info;
  864. int32_t dynamic_type = static_cast<int32_t>(FIXED);
  865. Status ret = GraphExecutor::GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  866. if (ret != SUCCESS) {
  867. GELOGE(ret, "Get dynamic input info failed.");
  868. return ret;
  869. }
  870. if (!batch_info.empty()) {
  871. SetDynamicInputDataFlag(run_input_data, batch_info, input_data);
  872. }
  873. }
  874. return GraphLoader::ExecuteModel(model_id, stream, async_mode, input_data, output_data);
  875. }
  876. /**
  877. * @ingroup ge
  878. * @brief Get weight memory size from model file
  879. * @param [in] const std::string &path: Offline model file path
  880. * @param [out] size_t &mem_size Execution memory size
  881. size_t &weight_size Weight memory space size
  882. * @return SUCCESS handle successfully / others handle failed
  883. */
  884. Status GeExecutor::GetMemAndWeightSize(const std::string &path, size_t &mem_size, size_t &weight_size) {
  885. GELOGI("Get memory and weight size from file begin.");
  886. if (!isInit_) {
  887. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  888. return ACL_ERROR_GE_EXEC_NOT_INIT;
  889. }
  890. ModelData model;
  891. std::string key;
  892. Status ret = ge::GraphLoader::LoadDataFromFile(path, key, 0, model);
  893. if ((ret != SUCCESS) || (model.model_data == nullptr)) {
  894. GELOGE(ret, "Load data from file failed. ret = %d", ret);
  895. return ret;
  896. }
  897. ret = ge::ModelManager::GetModelMemAndWeightSize(model, mem_size, weight_size);
  898. delete[] static_cast<char *>(model.model_data);
  899. model.model_data = nullptr;
  900. return ret;
  901. }
  902. /**
  903. * @ingroup ge
  904. * @brief Get weight memory size from model file
  905. * @param [in] const void *model_data Offline model buffer
  906. size_t model_size Offline model buffer length
  907. * @param [out] size_t &mem_size Execution memory size
  908. size_t &weight_size Weight memory space size
  909. * @return SUCCESS handle successfully / others handle failed
  910. */
  911. Status GeExecutor::GetMemAndWeightSize(const void *model_data, size_t model_size, size_t &mem_size,
  912. size_t &weight_size) {
  913. GELOGI("Get memory and weight size from data begin.");
  914. if (!isInit_) {
  915. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "GeExecutor has not been initialized!");
  916. return ACL_ERROR_GE_EXEC_NOT_INIT;
  917. }
  918. if (model_data == nullptr) {
  919. GELOGE(ACL_ERROR_GE_EXEC_MODEL_ADDR_INVALID, "invalid model data!");
  920. return ACL_ERROR_GE_EXEC_MODEL_ADDR_INVALID;
  921. }
  922. ModelData model;
  923. model.model_data = const_cast<void *>(model_data);
  924. model.model_len = static_cast<uint32_t>(model_size);
  925. return ge::ModelManager::GetModelMemAndWeightSize(model, mem_size, weight_size);
  926. }
  927. Status GeExecutor::LoadSingleOp(const std::string &model_name, const ge::ModelData &modelData, void *stream,
  928. SingleOp **single_op) {
  929. return SingleOpManager::GetInstance().GetOpFromModel(model_name, modelData, stream, single_op);
  930. }
  931. Status GeExecutor::LoadDynamicSingleOp(const std::string &model_name, const ge::ModelData &modelData, void *stream,
  932. DynamicSingleOp **single_op) {
  933. return SingleOpManager::GetInstance().GetDynamicOpFromModel(model_name, modelData, stream, single_op);
  934. }
  935. Status GeExecutor::ExecuteAsync(SingleOp *executor, const std::vector<DataBuffer> &inputs,
  936. std::vector<DataBuffer> &outputs) {
  937. if (executor == nullptr) {
  938. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "param is NULL");
  939. return ACL_ERROR_GE_EXEC_NOT_INIT;
  940. }
  941. return executor->ExecuteAsync(inputs, outputs);
  942. }
  943. ge::Status GeExecutor::ExecuteAsync(DynamicSingleOp *executor, const vector<GeTensorDesc> &input_desc,
  944. const vector<DataBuffer> &inputs, vector<GeTensorDesc> &output_desc,
  945. vector<DataBuffer> &outputs) {
  946. GE_CHECK_NOTNULL(executor);
  947. return executor->ExecuteAsync(input_desc, inputs, output_desc, outputs);
  948. }
  949. Status GeExecutor::ReleaseSingleOpResource(void *stream) {
  950. ModelManager::GetInstance()->ClearAicpuSo();
  951. return SingleOpManager::GetInstance().ReleaseResource(stream);
  952. }
  953. Status GeExecutor::GetDeviceIdByModelId(uint32_t model_id, uint32_t &device_id) {
  954. auto model_manager = ModelManager::GetInstance();
  955. GE_CHECK_NOTNULL(model_manager);
  956. auto davinci_model = model_manager->GetModel(model_id);
  957. if (davinci_model == nullptr) {
  958. GELOGE(ACL_ERROR_GE_EXEC_MODEL_ID_INVALID, "Model id: %d is invaild or model is not loaded.", model_id);
  959. return ACL_ERROR_GE_EXEC_MODEL_ID_INVALID;
  960. }
  961. device_id = davinci_model->GetDeviceId();
  962. return SUCCESS;
  963. }
  964. Status GeExecutor::GetBatchInfoSize(uint32_t model_id, size_t &shape_count) {
  965. std::vector<std::vector<int64_t>> batch_info;
  966. int32_t dynamic_type = static_cast<int32_t>(FIXED);
  967. Status ret = GetDynamicBatchInfo(model_id, batch_info, dynamic_type);
  968. if (ret != SUCCESS) {
  969. GELOGE(ret, "Calc batch info size failed. ret = %d", ret);
  970. return ret;
  971. }
  972. if (batch_info.empty()) {
  973. shape_count = kStaticBatchInfoSize;
  974. } else {
  975. shape_count = batch_info.size();
  976. }
  977. return SUCCESS;
  978. }
  979. Status GeExecutor::GetOrigInputInfo(uint32_t model_id, uint32_t index, OriginInputInfo &orig_input_info) {
  980. GELOGI("Begin to GetOrigInputInfo.");
  981. if (!isInit_) {
  982. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  983. return ACL_ERROR_GE_EXEC_NOT_INIT;
  984. }
  985. Status ret = GraphExecutor::GetOrigInputInfo(model_id, index, orig_input_info);
  986. if (ret != SUCCESS) {
  987. GELOGE(ret, "GetOrigInputInfo failed.");
  988. return ret;
  989. }
  990. GELOGI("GetOrigInputInfo succ.");
  991. return SUCCESS;
  992. }
  993. Status GeExecutor::GetAllAippInputOutputDims(uint32_t model_id, uint32_t index,
  994. std::vector<InputOutputDims> &input_dims,
  995. std::vector<InputOutputDims> &output_dims) {
  996. GELOGI("Begin to GetAllAippInputOutputDims.");
  997. if (!isInit_) {
  998. GELOGE(ACL_ERROR_GE_EXEC_NOT_INIT, "not inited yet!");
  999. return ACL_ERROR_GE_EXEC_NOT_INIT;
  1000. }
  1001. Status ret = GraphExecutor::GetAllAippInputOutputDims(model_id, index, input_dims, output_dims);
  1002. if (ret != SUCCESS) {
  1003. GELOGE(ret, "GetAllAippInputOutputDims failed.");
  1004. return ret;
  1005. }
  1006. GELOGI("GetAllAippInputOutputDims succ.");
  1007. return SUCCESS;
  1008. }
  1009. Status GeExecutor::GetOpDescInfo(uint32_t device_id, uint32_t stream_id, uint32_t task_id, OpDescInfo &op_desc_info) {
  1010. GELOGI("Begin to GetOpDescInfo.");
  1011. Status ret = GraphExecutor::GetOpDescInfo(device_id, stream_id, task_id, op_desc_info);
  1012. if (ret != SUCCESS) {
  1013. GELOGE(ret, "GetOpDescInfo failed.");
  1014. return ret;
  1015. }
  1016. GELOGI("GetOpDescInfo succ.");
  1017. return SUCCESS;
  1018. }
  1019. Status GeExecutor::SetDump(const DumpConfig &dump_config) {
  1020. GELOGI("Start to set dump config");
  1021. auto ret = DumpManager::GetInstance().SetDumpConf(dump_config);
  1022. if (ret != SUCCESS) {
  1023. GELOGE(ret, "Set dump conf failed");
  1024. return ret;
  1025. }
  1026. GELOGI("Set dump config successfully");
  1027. return SUCCESS;
  1028. }
  1029. } // namespace ge

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