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.

aicpu_task_struct.h 2.5 kB

5 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  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 AICPU_TASK_STRUCT_H_
  17. #define AICPU_TASK_STRUCT_H_
  18. #include <cstdint>
  19. namespace aicpu {
  20. #pragma pack(push, 1)
  21. struct AicpuParamHead
  22. {
  23. uint32_t length; // Total length: include cunstom message
  24. uint32_t ioAddrNum; // Input and output address number
  25. uint32_t extInfoLength; // extInfo struct Length
  26. uint64_t extInfoAddr; // extInfo address
  27. };
  28. enum class AicpuConfigMsgType {
  29. AICPU_CONFIG_MSG_TYPE_BUF_FREE = 0, /* free buf */
  30. AICPU_CONFIG_MSG_TYPE_BUF_RESET = 1, /* reset buf */
  31. AICPU_CONFIG_MSG_TYPE_BUF_SET_ADDR = 2, /* set buf addr to aicpu */
  32. };
  33. enum class AicpuErrMsgType {
  34. ERR_MSG_TYPE_NULL = 0,
  35. ERR_MSG_TYPE_AICORE = 1,
  36. ERR_MSG_TYPE_AICPU = 2,
  37. };
  38. typedef struct tagAicpuConfigMsg {
  39. uint8_t msgType;
  40. uint8_t reserved1;
  41. uint16_t bufLen;
  42. uint32_t offset;
  43. uint64_t bufAddr;
  44. uint32_t tsId;
  45. uint32_t reserved2;
  46. } AicpuConfigMsg;
  47. typedef struct tagAicoreErrMsgInfo {
  48. uint8_t errType;
  49. uint8_t version;
  50. uint8_t reserved1[2]; /* reserved1, 4 byte alignment */
  51. uint32_t errorCode;
  52. uint32_t modelId;
  53. uint32_t taskId;
  54. uint32_t streamId;
  55. uint64_t transactionId;
  56. uint8_t reserved2[228]; /* the total byte is 256, reserved2 len = 256 - other lens */
  57. } AicoreErrMsgInfo;
  58. typedef struct tagAicpuErrMsgInfo {
  59. uint8_t errType;
  60. uint8_t version;
  61. uint8_t reserved1[2]; /* reserved1, 4 byte alignment */
  62. uint32_t errorCode;
  63. uint32_t modelId;
  64. uint32_t streamId;
  65. uint64_t transactionId;
  66. char opName[64]; /* op name str */
  67. char errDesc[128]; /* err msg desc info */
  68. uint8_t reserved2[40]; /* the total byte is 256, reserved2 len = 256 - other lens */
  69. } AicpuErrMsgInfo;
  70. #pragma pack(pop)
  71. } // namespace aicpu
  72. #endif // AICPU_TASK_STRUCT_H_

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