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_api_error_codes.h 2.5 kB

5 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. /**
  2. * Copyright 2019-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. #ifndef INC_EXTERNAL_GE_GE_API_ERROR_CODES_H_
  17. #define INC_EXTERNAL_GE_GE_API_ERROR_CODES_H_
  18. #include <map>
  19. #include <string>
  20. namespace ge {
  21. class StatusFactory {
  22. public:
  23. static StatusFactory *Instance() {
  24. static StatusFactory instance;
  25. return &instance;
  26. }
  27. void RegisterErrorNo(uint32_t err, const std::string &desc) {
  28. // Avoid repeated addition
  29. if (err_desc_.find(err) != err_desc_.end()) {
  30. return;
  31. }
  32. err_desc_[err] = desc;
  33. }
  34. std::string GetErrDesc(uint32_t err) {
  35. auto iter_find = err_desc_.find(err);
  36. if (iter_find == err_desc_.end()) {
  37. return "";
  38. }
  39. return iter_find->second;
  40. }
  41. protected:
  42. StatusFactory() {}
  43. ~StatusFactory() {}
  44. private:
  45. std::map<uint32_t, std::string> err_desc_;
  46. };
  47. class ErrorNoRegisterar {
  48. public:
  49. ErrorNoRegisterar(uint32_t err, const std::string &desc) { StatusFactory::Instance()->RegisterErrorNo(err, desc); }
  50. ~ErrorNoRegisterar() {}
  51. };
  52. // Code compose(4 byte), runtime: 2 bit, type: 2 bit, level: 3 bit, sysid: 8 bit, modid: 5 bit, value: 12 bit
  53. #define GE_ERRORNO(runtime, type, level, sysid, modid, name, value, desc) \
  54. constexpr ge::Status name = \
  55. ((0xFF & (static_cast<uint8_t>(runtime))) << 30) | ((0xFF & (static_cast<uint8_t>(type))) << 28) | \
  56. ((0xFF & (static_cast<uint8_t>(level))) << 25) | ((0xFF & (static_cast<uint8_t>(sysid))) << 17) | \
  57. ((0xFF & (static_cast<uint8_t>(modid))) << 12) | (0x0FFF & (static_cast<uint16_t>(value))); \
  58. const ErrorNoRegisterar g_##name##_errorno(name, desc);
  59. using Status = uint32_t;
  60. // General error code
  61. GE_ERRORNO(0, 0, 0, 0, 0, SUCCESS, 0, "success");
  62. GE_ERRORNO(0b11, 0b11, 0b111, 0xFF, 0b11111, FAILED, 0xFFF, "failed");
  63. } // namespace ge
  64. #endif // INC_EXTERNAL_GE_GE_API_ERROR_CODES_H_

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

Contributors (1)