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_ir_build.cc 24 kB

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
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561
  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 "external/ge/ge_ir_build.h"
  17. #include <vector>
  18. #include "common/auth/file_saver.h"
  19. #include "external/register/register_types.h"
  20. #include "framework/common/debug/ge_log.h"
  21. #include "framework/common/ge_inner_error_codes.h"
  22. #include "framework/common/string_util.h"
  23. #include "framework/common/types.h"
  24. #include "framework/common/util.h"
  25. #include "framework/omg/omg_inner_types.h"
  26. #include "framework/omg/omg_inner_types.h"
  27. #include "ge/ge_api_types.h"
  28. #include "generator/ge_generator.h"
  29. #include "graph/compute_graph.h"
  30. #include "graph/ge_tensor.h"
  31. #include "graph/utils/type_utils.h"
  32. #include "graph/ge_global_options.h"
  33. #include "init/gelib.h"
  34. #include "ir_build/atc_ir_common.h"
  35. #include "model/ge_model.h"
  36. #include "graph/shape_refiner.h"
  37. #include "graph/opsproto_manager.h"
  38. using std::string;
  39. using namespace std;
  40. namespace ge {
  41. namespace {
  42. const std::string IR_OPTION_TARGET = "target";
  43. const std::string IR_OPTION_MODE = "mode";
  44. const std::string IR_OP_CONF_DELIMITER = ":";
  45. const std::string IR_OPTION_LOG_LEVEL_DEFAULT = "default";
  46. const std::string IR_OPTION_BUFFER_OPTIMIZE_DEFAULT = "l2_optimize";
  47. const std::string IR_OPTION_DISABLE_REUSE_MEMORY_DEFAULT = "0";
  48. const std::string IR_OPTION_ENABLE_COMPRESS_WEIGHT_DEFAULT = "false";
  49. } // namespace
  50. static graphStatus CheckGlobalOptions(std::map<std::string, std::string> &global_options) {
  51. // check param disable_reuse_memory
  52. std::string disable_reuse_memory = global_options.find(ge::ir_option::EXEC_DISABLE_REUSED_MEMORY) ==
  53. global_options.end()
  54. ? IR_OPTION_DISABLE_REUSE_MEMORY_DEFAULT
  55. : global_options[ge::ir_option::EXEC_DISABLE_REUSED_MEMORY];
  56. GE_CHK_BOOL_EXEC(ge::CheckDisableReuseMemoryParamValid(disable_reuse_memory) == ge::SUCCESS,
  57. return ge::GRAPH_PARAM_INVALID, "check disable_reuse_memory failed!");
  58. global_options[ge::ir_option::EXEC_DISABLE_REUSED_MEMORY] = disable_reuse_memory;
  59. // check buffer_optimize
  60. std::string buffer_optimize = global_options.find(ge::ir_option::BUFFER_OPTIMIZE) == global_options.end()
  61. ? IR_OPTION_BUFFER_OPTIMIZE_DEFAULT
  62. : global_options[ge::ir_option::BUFFER_OPTIMIZE];
  63. GE_CHK_BOOL_EXEC(ge::CheckBufferOptimizeParamValid(buffer_optimize) == ge::SUCCESS,
  64. return ge::GRAPH_PARAM_INVALID, "check buffer optimize failed!");
  65. global_options[ge::ir_option::BUFFER_OPTIMIZE] = buffer_optimize;
  66. // check enable_single_stream
  67. std::string enable_single_stream = global_options.find(ge::ir_option::ENABLE_SINGLE_STREAM) == global_options.end()
  68. ? ""
  69. : global_options[ge::ir_option::ENABLE_SINGLE_STREAM];
  70. GE_CHK_BOOL_EXEC(ge::CheckEnableSingleStreamParamValid(enable_single_stream) == ge::SUCCESS,
  71. return ge::GRAPH_PARAM_INVALID, "check enable single stream failed!");
  72. // check compress_weight
  73. std::string enable_compress_weight = global_options.find(ge::ir_option::ENABLE_COMPRESS_WEIGHT) ==
  74. global_options.end()
  75. ? IR_OPTION_ENABLE_COMPRESS_WEIGHT_DEFAULT
  76. : global_options[ge::ir_option::ENABLE_COMPRESS_WEIGHT];
  77. std::string compress_weight_conf = global_options.find(ge::ir_option::COMPRESS_WEIGHT_CONF) == global_options.end()
  78. ? ""
  79. : global_options[ge::ir_option::COMPRESS_WEIGHT_CONF];
  80. GE_CHK_BOOL_EXEC(ge::CheckCompressWeightParamValid(enable_compress_weight, compress_weight_conf) == ge::SUCCESS,
  81. return ge::GRAPH_PARAM_INVALID, "check compress weight failed!");
  82. global_options[ge::ir_option::ENABLE_COMPRESS_WEIGHT] = (enable_compress_weight == "true") ?
  83. ge::kEnableCompressWeightTrue :
  84. ge::kEnableCompressWeightFalse;
  85. // check optypelist_for_implmode and op_select_implmode
  86. std::string optypelist_for_implmode = global_options.find(ge::ir_option::OPTYPELIST_FOR_IMPLMODE) ==
  87. global_options.end()
  88. ? ""
  89. : global_options[ge::ir_option::OPTYPELIST_FOR_IMPLMODE];
  90. std::string op_select_implmode = global_options.find(ge::ir_option::OP_SELECT_IMPL_MODE) ==
  91. global_options.end()
  92. ? ""
  93. : global_options[ge::ir_option::OP_SELECT_IMPL_MODE];
  94. GE_CHK_BOOL_EXEC(
  95. ge::CheckImplmodeParamValid(optypelist_for_implmode, op_select_implmode) == ge::SUCCESS,
  96. return ge::GRAPH_PARAM_INVALID, "check optypelist_for_implmode and op_select_implmode failed!");
  97. global_options[ge::ir_option::OP_SELECT_IMPL_MODE] = op_select_implmode;
  98. // set precision mode default value
  99. std::string precision_mode = global_options.find(ge::ir_option::PRECISION_MODE) ==
  100. global_options.end()
  101. ? "force_fp16"
  102. : global_options[ge::ir_option::PRECISION_MODE];
  103. global_options[ge::ir_option::PRECISION_MODE] = precision_mode;
  104. return GRAPH_SUCCESS;
  105. }
  106. static void GetOpsProtoPath(string &opsproto_path) {
  107. GELOGI("Start to get ops proto path schedule.");
  108. const char *path_env = std::getenv("ASCEND_OPP_PATH");
  109. if (path_env != nullptr) {
  110. string path = path_env;
  111. string file_path = RealPath(path.c_str());
  112. if (file_path.empty()) {
  113. GELOGE(FAILED, "File path %s is invalid.", path.c_str());
  114. return;
  115. }
  116. opsproto_path = (path + "/op_proto/custom/" + ":") + (path + "/op_proto/built-in/");
  117. GELOGI("Get opsproto so path from env : %s", path.c_str());
  118. return;
  119. }
  120. string path_base = PluginManager::GetPath();
  121. GELOGI("path_base is %s", path_base.c_str());
  122. path_base = path_base.substr(0, path_base.rfind('/'));
  123. path_base = path_base.substr(0, path_base.rfind('/') + 1);
  124. opsproto_path = (path_base + "ops/op_proto/custom/" + ":") + (path_base + "ops/op_proto/built-in/");
  125. }
  126. static void LoadOpsProto() {
  127. string opsproto_path;
  128. GetOpsProtoPath(opsproto_path);
  129. GELOGI("Get opsproto path is %s", opsproto_path.c_str());
  130. OpsProtoManager *manager = OpsProtoManager::Instance();
  131. map<string, string> option_tmp;
  132. option_tmp.emplace(std::pair<string, string>(string("ge.opsProtoLibPath"), opsproto_path));
  133. (void)manager->Initialize(option_tmp);
  134. }
  135. graphStatus aclgrphBuildInitialize(std::map<std::string, std::string> global_options) {
  136. GELOGD("Enter aclgrphInitialize start!");
  137. // check global options
  138. if (CheckGlobalOptions(global_options) != GRAPH_SUCCESS) {
  139. GELOGE(GRAPH_PARAM_INVALID, "Check global options falied!");
  140. return GRAPH_PARAM_INVALID;
  141. }
  142. // print global option map
  143. ge::PrintOptionMap(global_options, "global option");
  144. LoadOpsProto();
  145. std::shared_ptr<ge::GELib> instance_ptr = ge::GELib::GetInstance();
  146. if (instance_ptr == nullptr || !instance_ptr->InitFlag()) {
  147. GELOGI("aclgrphInitialize start!");
  148. auto ret = ge::GELib::Initialize(global_options);
  149. if (ret != ge::SUCCESS) {
  150. GELOGE(ret, "GE initialize failed!");
  151. return GRAPH_FAILED;
  152. }
  153. }
  154. GELOGW("gelib has been initialized!");
  155. return GRAPH_SUCCESS;
  156. }
  157. void aclgrphBuildFinalize() {
  158. if (ge::GELib::GetInstance() != nullptr && ge::GELib::GetInstance()->InitFlag()) {
  159. (void)ge::GELib::GetInstance()->Finalize();
  160. return;
  161. }
  162. GELOGW("[Notice] gelib has not been initialized!do nothing!");
  163. }
  164. class Impl {
  165. public:
  166. Impl() {
  167. omg_context_ = domi::GetContext();
  168. omg_context_.format = domi::DOMI_TENSOR_ND;
  169. omg_context_.input_nodes_format_map.clear();
  170. omg_context_.output_formats.clear();
  171. omg_context_.user_input_dims.clear();
  172. omg_context_.input_dims.clear();
  173. omg_context_.op_conf_map.clear();
  174. omg_context_.out_nodes_map.clear();
  175. omg_context_.user_out_nodes.clear();
  176. omg_context_.net_format = domi::DOMI_TENSOR_RESERVED;
  177. omg_context_.type = domi::FRAMEWORK_RESERVED;
  178. omg_context_.run_mode = ONLY_PRE_CHECK;
  179. omg_context_.train_flag = false;
  180. omg_context_.output_type.clear();
  181. omg_context_.is_dynamic_input = false;
  182. omg_context_.dynamic_batch_size.clear();
  183. omg_context_.dynamic_image_size.clear();
  184. omg_context_.dynamic_dims.clear();
  185. };
  186. ~Impl() { (void)generator_.Finalize(); };
  187. graphStatus CheckOptions(const std::map<std::string, std::string> &options);
  188. graphStatus CreateInputsForIRBuild(const ge::Graph &graph, vector<ge::GeTensor> &inputs);
  189. graphStatus Init(const std::map<std::string, std::string> &options);
  190. graphStatus BuildModel(const Graph &graph, const std::map<std::string, std::string> &options,
  191. ModelBufferData &ge_models);
  192. graphStatus InitDomiOmgContext(const string &input_shape, const string &input_format, const string &net_format,
  193. bool is_dynamic_input);
  194. void SetRtSocVersion();
  195. void UpdateThreadContext();
  196. void LoadOpsProto();
  197. public:
  198. ge::GeGenerator generator_;
  199. std::map<std::string, std::string> options_;
  200. bool is_dynamic_input_ = false;
  201. OmgContext omg_context_;
  202. };
  203. graphStatus Impl::CheckOptions(const std::map<std::string, std::string> &options) {
  204. for (auto &ele : options) {
  205. auto it = ge::ir_option::ir_builder_suppported_options.find(ele.first);
  206. if (it == ge::ir_option::ir_builder_suppported_options.end()) {
  207. auto it_lx_fusion = ir_builder_supported_options_for_lx_fusion.find(ele.first);
  208. if (it_lx_fusion == ir_builder_supported_options_for_lx_fusion.end()) {
  209. GELOGE(GRAPH_PARAM_INVALID, "input options include unsupported option(%s).Please check!",
  210. ele.first.c_str());
  211. return GRAPH_PARAM_INVALID;
  212. }
  213. }
  214. options_.insert(ele);
  215. }
  216. // Check options build_mode and build_step.
  217. std::string build_mode;
  218. auto it = options_.find(BUILD_MODE);
  219. if (it != options_.end() && !(it->second.empty())) {
  220. if (build_mode_options.find(it->second) == build_mode_options.end()) {
  221. GELOGE(GRAPH_PARAM_INVALID, "Build mode:%s is unsupported. Please check!", it->second.c_str());
  222. return GRAPH_PARAM_INVALID;
  223. }
  224. build_mode = it->second;
  225. }
  226. it = options_.find(BUILD_STEP);
  227. if (it != options_.end() && !(it->second.empty())) {
  228. if (build_step_options.find(it->second) == build_step_options.end()) {
  229. GELOGE(GRAPH_PARAM_INVALID, "Build step:%s is unsupported. Please check!", it->second.c_str());
  230. return GRAPH_PARAM_INVALID;
  231. }
  232. } else {
  233. if (build_mode == BUILD_MODE_TUNING) {
  234. GELOGE(GRAPH_PARAM_INVALID, "Build mode tuning must specify build step. Please check!");
  235. return GRAPH_PARAM_INVALID;
  236. }
  237. }
  238. return GRAPH_SUCCESS;
  239. }
  240. graphStatus Impl::Init(const std::map<std::string, std::string> &options) {
  241. // 1. check options
  242. graphStatus ret = CheckOptions(options);
  243. if (ret != GRAPH_SUCCESS) {
  244. GELOGE(ret, "User input options are illegal! Please check!");
  245. return ret;
  246. }
  247. std::string build_mode = (options_.find(BUILD_MODE) == options_.end() || options_[BUILD_MODE] == BUILD_MODE_NORMAL)
  248. ? "" : options_[BUILD_MODE];
  249. options_[BUILD_MODE] = build_mode;
  250. // set log level
  251. std::string log = options_.find(ge::ir_option::LOG_LEVEL) == options_.end()
  252. ? IR_OPTION_LOG_LEVEL_DEFAULT
  253. : options_[ge::ir_option::LOG_LEVEL];
  254. GE_CHK_BOOL_RET_STATUS_NOLOG(ge::CheckLogParamValidAndSetLogLevel(log) == 0, GRAPH_PARAM_INVALID);
  255. options_[ge::ir_option::LOG_LEVEL] = log;
  256. string input_shape = options_.find("input_shape") == options_.end() ? "" : options_["input_shape"];
  257. string input_format = options_.find("input_format") == options_.end() ? "" : options_["input_format"];
  258. string net_format = options_.find("net_format") == options_.end() ? "" : options_["net_format"];
  259. string dynamic_batch_size = options_.find(ge::ir_option::DYNAMIC_BATCH_SIZE) == options_.end()
  260. ? ""
  261. : options_[ge::ir_option::DYNAMIC_BATCH_SIZE];
  262. string dynamic_image_size = options_.find(ge::ir_option::DYNAMIC_IMAGE_SIZE) == options_.end()
  263. ? ""
  264. : options_[ge::ir_option::DYNAMIC_IMAGE_SIZE];
  265. string dynamic_dims =
  266. options_.find(ge::ir_option::DYNAMIC_DIMS) == options_.end() ? "" : options_[ge::ir_option::DYNAMIC_DIMS];
  267. auto status = CheckDynamicInputParamValid(dynamic_batch_size, dynamic_image_size, dynamic_dims, input_shape,
  268. input_format, is_dynamic_input_);
  269. if (status != ge::SUCCESS) {
  270. GELOGE(GRAPH_PARAM_INVALID, "Check dynamic input size failed!");
  271. return GRAPH_PARAM_INVALID;
  272. }
  273. GELOGD("User input dynamic_batch_size:%s, dynamic_image_size:%s, dynamic_dims:%s.", dynamic_batch_size.c_str(),
  274. dynamic_image_size.c_str(), dynamic_dims.c_str());
  275. omg_context_.dynamic_batch_size = dynamic_batch_size;
  276. omg_context_.dynamic_image_size = dynamic_image_size;
  277. omg_context_.dynamic_dims = dynamic_dims;
  278. // check output_type
  279. std::string output_type = options_.find(ge::ir_option::OUTPUT_TYPE) == options_.end()
  280. ? ""
  281. : options_[ge::ir_option::OUTPUT_TYPE];
  282. GE_CHK_BOOL_EXEC(ge::CheckOutputTypeParamValid(output_type) == ge::SUCCESS,
  283. return ge::GRAPH_PARAM_INVALID, "check output type failed!");
  284. // check insert_op_conf
  285. std::string insert_op_conf = options_.find(ge::ir_option::INSERT_OP_FILE) == options_.end()
  286. ? ""
  287. : options_[ge::ir_option::INSERT_OP_FILE];
  288. GE_CHK_BOOL_EXEC(ge::CheckInsertOpConfParamValid(std::string(insert_op_conf)) == ge::SUCCESS,
  289. return ge::GRAPH_PARAM_INVALID, "check insert op conf failed!");
  290. GE_CHK_BOOL_EXEC(insert_op_conf.empty() || dynamic_dims.empty(),
  291. return ge::GRAPH_PARAM_INVALID, "dynamic dims function does not support aipp");
  292. // for IR builder.Only support om mode, so here fixed;
  293. options_.insert(std::pair<string, string>(string(IR_OPTION_MODE), to_string(0)));
  294. options_.insert(std::pair<string, string>(string(IR_OPTION_TARGET), "mini"));
  295. options_.insert(std::pair<string, string>(string(ge::RUN_FLAG), to_string(0)));
  296. options_.insert(std::pair<string, string>(string(ge::TRAIN_FLAG), to_string(0)));
  297. options_.insert(std::pair<string, string>(string(ge::SAVE_ORIGINAL_MODEL), to_string(0)));
  298. // print ge option map
  299. ge::PrintOptionMap(options_, "ge option");
  300. SetRtSocVersion();
  301. UpdateThreadContext();
  302. // 3. init generator with options_
  303. ret = generator_.Initialize(options_, omg_context_);
  304. if (ret != GRAPH_SUCCESS) {
  305. GELOGE(ret, "generator Initialize failed!");
  306. return ret;
  307. }
  308. // 4.parse and init Context with input shape format and net format info
  309. return this->InitDomiOmgContext(input_shape, input_format, net_format, is_dynamic_input_);
  310. }
  311. void Impl::SetRtSocVersion() {
  312. const auto &global_options = GetMutableGlobalOptions();
  313. auto it = global_options.find(ge::SOC_VERSION);
  314. if (it != global_options.end()) {
  315. const char *soc_version = it->second.c_str();
  316. rtError_t rt_ret = rtSetSocVersion(soc_version);
  317. if (rt_ret != RT_ERROR_NONE) {
  318. GELOGW("Set soc version %s failed. ret:0x%X", soc_version, rt_ret);
  319. }
  320. GELOGD("Set soc version %s success.", soc_version);
  321. }
  322. }
  323. void Impl::UpdateThreadContext() {
  324. GetThreadLocalContext().SetGlobalOption(GetMutableGlobalOptions());
  325. GetThreadLocalContext().SetGraphOption(options_);
  326. }
  327. graphStatus Impl::CreateInputsForIRBuild(const ge::Graph &graph, vector<ge::GeTensor> &inputs) {
  328. auto compute_graph = ge::GraphUtils::GetComputeGraph(graph);
  329. GE_CHECK_NOTNULL(compute_graph);
  330. int64_t index = 0;
  331. for (ge::NodePtr &input_node : compute_graph->GetDirectNode()) {
  332. GE_CHECK_NOTNULL(input_node);
  333. ge::OpDescPtr op = input_node->GetOpDesc();
  334. GE_CHECK_NOTNULL(op);
  335. if (op->GetType() == DATA) {
  336. (void)AttrUtils::SetInt(op, ATTR_NAME_INDEX, index++);
  337. GELOGD("Data op inputDesc size: %zu", op->GetAllInputsDesc().size());
  338. ge::GeTensorDesc tensor = op->GetInputDesc(0);
  339. string data_op_name = op->GetName();
  340. GELOGD("Data op name: %s", data_op_name.c_str());
  341. ge::GeShape data_shape;
  342. auto iter = omg_context_.input_dims.find(data_op_name);
  343. if (iter != omg_context_.input_dims.end()) {
  344. data_shape = ge::GeShape(iter->second);
  345. GELOGD("Data op get shape from Context.");
  346. } else {
  347. data_shape = tensor.GetShape();
  348. GELOGD("Data op get shape from InputDesc in ge ir graph.");
  349. }
  350. // If user point input format, do work for all data ops; else do according to tensor_desc
  351. auto data_format = omg_context_.format != domi::DOMI_TENSOR_ND ?
  352. ge::TypeUtils::DomiFormatToFormat(omg_context_.format) : tensor.GetFormat();
  353. ge::DataType data_type = tensor.GetDataType();
  354. string data_type_str = ge::TypeUtils::DataTypeToSerialString(data_type);
  355. GELOGD("Data op get data type:%s from InputDesc in ge ir graph.", data_type_str.c_str());
  356. ge::GeTensor inputTensor;
  357. ge::GeTensorDesc desc(data_shape, ge::Format(data_format), data_type);
  358. inputTensor.SetTensorDesc(desc);
  359. inputs.push_back(inputTensor);
  360. }
  361. }
  362. GELOGD("CreateInputsForIRBuild, inputs size: %zu", inputs.size());
  363. return GRAPH_SUCCESS;
  364. }
  365. graphStatus Impl::BuildModel(const Graph &graph, const std::map<std::string, std::string> &options,
  366. ModelBufferData &model) {
  367. // 1. init GeGenerator with user optios
  368. graphStatus ret = Init(options);
  369. if (ret != GRAPH_SUCCESS) {
  370. GELOGE(ret, "Build ir model Init failed!");
  371. return ret;
  372. }
  373. // 2. construct input
  374. std::vector<GeTensor> inputs;
  375. if (!omg_context_.is_dynamic_input) { // if dynamic input , no need to creat inputs
  376. ret = CreateInputsForIRBuild(graph, inputs);
  377. if (ret != GRAPH_SUCCESS) {
  378. GELOGE(ret, "CreateInputsForIRBuild failed!");
  379. return ret;
  380. }
  381. }
  382. // 3. build IR model
  383. ret = generator_.GenerateOnlineModel(graph, inputs, model);
  384. if (ret != GRAPH_SUCCESS) {
  385. GELOGE(ret, "GenerateOnlineModel failed!");
  386. return ret;
  387. }
  388. return GRAPH_SUCCESS;
  389. }
  390. graphStatus Impl::InitDomiOmgContext(const string &input_shape, const string &input_format, const string &net_format,
  391. bool is_dynamic_input) {
  392. // Clear omgcontext data first
  393. omg_context_.input_dims.clear();
  394. omg_context_.user_input_dims.clear();
  395. omg_context_.is_dynamic_input = is_dynamic_input;
  396. // the default value is ND
  397. omg_context_.format = domi::DOMI_TENSOR_ND;
  398. if (!input_format.empty()) {
  399. auto iter = ge::input_format_str_to_geformat.find(input_format);
  400. if (iter != ge::input_format_str_to_geformat.end()) {
  401. omg_context_.format = iter->second;
  402. } else {
  403. GELOGE(GRAPH_PARAM_INVALID, "Input format %s not support , expect ND/NCHW/NHWC/CHWN/NC1HWC0/NHWC1C0.",
  404. input_format.c_str());
  405. return GRAPH_PARAM_INVALID;
  406. }
  407. }
  408. // Input is empty, do not process
  409. if (input_shape.empty()) {
  410. return GRAPH_SUCCESS;
  411. }
  412. if (!ParseInputShape(input_shape, omg_context_.input_dims, omg_context_.user_input_dims, is_dynamic_input)) {
  413. GELOGE(GRAPH_PARAM_INVALID, "Failed to parse input shape: %s", input_shape.c_str());
  414. return GRAPH_PARAM_INVALID;
  415. }
  416. return GRAPH_SUCCESS;
  417. }
  418. graphStatus aclgrphBuildModel(const ge::Graph &graph, const std::map<std::string, std::string> &build_options,
  419. ModelBufferData &model) {
  420. GELOGD("Enter aclmdlBuildModel process!");
  421. Impl builder;
  422. return builder.BuildModel(graph, build_options, model);
  423. }
  424. graphStatus aclgrphSaveModel(const string &output_file, const ModelBufferData &model) {
  425. GELOGD("Enter aclmdlSaveModel process!");
  426. if (model.data.get() == nullptr || model.length == 0) {
  427. GELOGE(GRAPH_PARAM_INVALID, "input model is illegal");
  428. return GRAPH_PARAM_INVALID;
  429. }
  430. return FileSaver::SaveToFile((output_file + ".om"), reinterpret_cast<void*>(model.data.get()),
  431. static_cast<uint32_t>(model.length));
  432. }
  433. graphStatus aclgrphGetIRVersion(int *major_version, int *minor_version, int *patch_version) {
  434. GELOGD("Enter aclgrphGetIRVersion process!");
  435. GE_CHECK_NOTNULL(major_version);
  436. GE_CHECK_NOTNULL(minor_version);
  437. GE_CHECK_NOTNULL(patch_version);
  438. *major_version = IR_MAJOR_VERSION;
  439. *minor_version = IR_MINOR_VERSION;
  440. *patch_version = IR_PATCH_VERSION;
  441. return GRAPH_SUCCESS;
  442. }
  443. graphStatus aclgrphInferShapeAndType(ge::Graph &graph) {
  444. auto compute_graph = GraphUtils::GetComputeGraph(graph);
  445. GE_CHECK_NOTNULL(compute_graph);
  446. auto root_graph = compute_graph->GetParentGraph();
  447. if (root_graph != nullptr) {
  448. GELOGE(GRAPH_PARAM_INVALID, "Input param should not be subgraph");
  449. return GRAPH_PARAM_INVALID;
  450. }
  451. auto ret = compute_graph->InferOriginFormat();
  452. if (ret != GRAPH_SUCCESS) {
  453. GELOGE(ret, "Acl InferOriginFormat failed.");
  454. return ret;
  455. }
  456. for (auto &node: compute_graph->GetAllNodes()) {
  457. graphStatus ret = ShapeRefiner::InferShapeAndType(node);
  458. if (ret == GRAPH_PARAM_INVALID) {
  459. GELOGW("Can not find infershape func.");
  460. continue;
  461. } else if (ret != GRAPH_SUCCESS) {
  462. GELOGE(ret, "Acl infershape failed.");
  463. return ret;
  464. }
  465. }
  466. return GRAPH_SUCCESS;
  467. }
  468. graphStatus aclgrphDumpGraph(const ge::Graph &graph, const char *file, const size_t len) {
  469. GE_CHECK_NOTNULL(file);
  470. if (len > PATH_MAX || len != strlen(file) || strlen(file) == 0) {
  471. GELOGE(GRAPH_PARAM_INVALID, "File path invalid.");
  472. return GRAPH_PARAM_INVALID;
  473. }
  474. auto compute_graph = GraphUtils::GetComputeGraph(graph);
  475. GE_CHECK_NOTNULL(compute_graph);
  476. string full_path(file, len);
  477. for (size_t i = 0; i < len; i++) {
  478. if (full_path[i] == '\\') {
  479. full_path.replace(i, 1, "/");
  480. }
  481. }
  482. string suffix;
  483. string file_path;
  484. int pos = full_path.rfind("/");
  485. if (pos != -1) {
  486. suffix = full_path.substr(pos + 1, -1);
  487. file_path = full_path.substr(0, pos);
  488. } else {
  489. suffix = full_path;
  490. file_path = "./";
  491. }
  492. if (suffix.empty()) {
  493. suffix = compute_graph->GetName();
  494. if (suffix.empty()) {
  495. suffix = "graph";
  496. }
  497. }
  498. char path[PATH_MAX] = {0};
  499. if (realpath(file_path.c_str(), path) == nullptr) {
  500. GELOGE(GRAPH_PARAM_INVALID, "Dump file path:%s is invalid.", file);
  501. return GRAPH_PARAM_INVALID;
  502. }
  503. GraphUtils::DumpGEGrph(compute_graph, string(path), suffix);
  504. GraphUtils::DumpGrphToOnnx(*compute_graph, string(path), suffix);
  505. uint64_t i = 0;
  506. for (const auto &sub_graph_func : compute_graph->GetAllSubgraphs()) {
  507. auto sub_graph_func_name = suffix + std::string("_sub_graph_") + std::to_string(i++);
  508. GraphUtils::DumpGEGrph(sub_graph_func, string(path), sub_graph_func_name);
  509. GraphUtils::DumpGrphToOnnx(*sub_graph_func, string(path), sub_graph_func_name);
  510. }
  511. return GRAPH_SUCCESS;
  512. }
  513. } // namespace ge

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