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.

base.h 8.2 kB

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
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
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  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 __CCE_RUNTIME_BASE_H__
  17. #define __CCE_RUNTIME_BASE_H__
  18. #include <stdint.h>
  19. #if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
  20. extern "C" {
  21. #endif
  22. // If you need export the function of this library in Win32 dll, use __declspec(dllexport)
  23. #ifndef RTS_API
  24. #ifdef RTS_DLL_EXPORT
  25. #define RTS_API __declspec(dllexport)
  26. #else
  27. #define RTS_API
  28. #endif
  29. #endif
  30. typedef int32_t rtError_t;
  31. static const int32_t RT_ERROR_NONE = 0; // success
  32. /**
  33. * @ingroup dvrt_base
  34. * @brief runtime exception numbers.
  35. */
  36. typedef enum tagRtExceptionType {
  37. RT_EXCEPTION_NONE = 0,
  38. RT_EXCEPTION_TS_DOWN = 1,
  39. RT_EXCEPTION_TASK_TIMEOUT = 2,
  40. RT_EXCEPTION_TASK_FAILURE = 3,
  41. RT_EXCEPTION_DEV_RUNNING_DOWN = 4,
  42. RT_EXCEPTION_STREAM_ID_FREE_FAILED = 5
  43. } rtExceptionType;
  44. /**
  45. * @ingroup dvrt_base
  46. * @brief Switch type.
  47. */
  48. typedef enum tagRtCondition {
  49. RT_EQUAL = 0,
  50. RT_NOT_EQUAL,
  51. RT_GREATER,
  52. RT_GREATER_OR_EQUAL,
  53. RT_LESS,
  54. RT_LESS_OR_EQUAL
  55. } rtCondition_t;
  56. /**
  57. * @ingroup dvrt_base
  58. * @brief Data Type of Extensible Switch Task.
  59. */
  60. typedef enum tagRtSwitchDataType {
  61. RT_SWITCH_INT32 = 0,
  62. RT_SWITCH_INT64 = 1,
  63. } rtSwitchDataType_t;
  64. typedef enum tagRtStreamFlagType {
  65. RT_HEAD_STREAM = 0, // first stream
  66. RT_INVALID_FLAG = 0xFFFFFFFF,
  67. } rtStreamFlagType_t;
  68. typedef enum tagRtLimitType {
  69. RT_LIMIT_TYPE_LOW_POWER_TIMEOUT = 0, // timeout for power down , ms
  70. } rtLimitType_t;
  71. typedef struct rtExceptionInfo {
  72. uint32_t taskid;
  73. uint32_t streamid;
  74. uint32_t tid;
  75. uint32_t deviceid;
  76. } rtExceptionInfo;
  77. typedef void (*rtErrorCallback)(rtExceptionType);
  78. typedef void (*rtTaskFailCallback)(rtExceptionInfo *exceptionInfo);
  79. typedef void (*rtDeviceStateCallback)(uint32_t devId, bool isOpen);
  80. /**
  81. * @ingroup dvrt_base
  82. * @brief stream handle.
  83. */
  84. typedef void *rtStream_t;
  85. /**
  86. * @ingroup dvrt_base
  87. * @brief runtime event handle.
  88. */
  89. typedef void *rtEvent_t;
  90. /**
  91. * @ingroup dvrt_base
  92. * @brief label handle.
  93. */
  94. typedef void *rtLabel_t;
  95. /**
  96. * @ingroup profiling_base
  97. * @brief runtime handle.
  98. */
  99. RTS_API rtError_t rtSetProfDirEx(const char *profDir, const char *address, const char *jobCtx);
  100. /**
  101. * @ingroup profiling_base
  102. * @brief init profiler object.
  103. */
  104. RTS_API rtError_t rtProfilerInit(const char *profDir, const char *address, const char *jobCtx);
  105. /**
  106. * @ingroup profiling_base
  107. * @brief config rts profiler.
  108. */
  109. RTS_API rtError_t rtProfilerConfig(uint16_t type);
  110. /**
  111. * @ingroup profiling_base
  112. * @brief start rts profiler.
  113. */
  114. RTS_API rtError_t rtProfilerStart(uint64_t profConfig, int32_t numsDev, uint32_t* deviceList);
  115. /**
  116. * @ingroup profiling_base
  117. * @brief stop rts profiler.
  118. */
  119. RTS_API rtError_t rtProfilerStop(uint64_t profConfig, int32_t numsDev, uint32_t* deviceList);
  120. /**
  121. * @ingroup profiling_base
  122. * @brief ts send keypoint profiler log.
  123. */
  124. RTS_API rtError_t rtProfilerTrace(uint64_t id, bool notify, uint32_t flags, rtStream_t stream);
  125. /**
  126. * @ingroup dvrt_base
  127. * @brief Returns the last error from a runtime call.
  128. */
  129. RTS_API rtError_t rtGetLastError();
  130. /**
  131. * @ingroup dvrt_base
  132. * @brief Returns the last error from a runtime call.
  133. */
  134. RTS_API rtError_t rtPeekAtLastError();
  135. /**
  136. * @ingroup dvrt_base
  137. * @brief register callback for error code
  138. * @param [out] NA
  139. * @return RT_ERROR_NONE for ok
  140. */
  141. RTS_API rtError_t rtSetExceptCallback(rtErrorCallback callback);
  142. /**
  143. * @ingroup dvrt_base
  144. * @brief register callback for task fail
  145. * @param [out] NA
  146. * @return RT_ERROR_NONE for ok
  147. */
  148. RTS_API rtError_t rtSetTaskFailCallback(rtTaskFailCallback callback);
  149. /**
  150. * @ingroup dvrt_base
  151. * @brief register callback for deviceid
  152. * @param [in] uniName unique register name, can't be null
  153. * @param [in] callback Device state callback function
  154. * @param [out] NA
  155. * @return RT_ERROR_NONE for ok
  156. */
  157. RTS_API rtError_t rtRegDeviceStateCallback(const char *regName, rtDeviceStateCallback callback);
  158. /**
  159. * @ingroup dvrt_base
  160. * @brief notify handle.
  161. */
  162. typedef void *rtNotify_t;
  163. /**
  164. * @ingroup dvrt_base
  165. * @brief create label instance
  166. * @param [out] label created label
  167. * @return RT_ERROR_NONE for ok
  168. * @return RT_ERROR_INVALID_VALUE for error input
  169. */
  170. RTS_API rtError_t rtLabelCreate(rtLabel_t *label);
  171. /**
  172. * @ingroup dvrt_base
  173. * @brief set label and stream instance
  174. * @param [in] label set label
  175. * @param [in] stream set stream
  176. * @return RT_ERROR_NONE for ok
  177. * @return RT_ERROR_INVALID_VALUE for error input
  178. */
  179. RTS_API rtError_t rtLabelSet(rtLabel_t label, rtStream_t stream);
  180. /**
  181. * @ingroup dvrt_base
  182. * @brief destroy label instance
  183. * @param [in] label label to destroy
  184. * @return RT_ERROR_NONE for ok
  185. * @return RT_ERROR_INVALID_VALUE for error input
  186. */
  187. RTS_API rtError_t rtLabelDestroy(rtLabel_t label);
  188. /**
  189. * @ingroup dvrt_base
  190. * @brief label switch instance
  191. * @param [in] ptr address to get value compared
  192. * @param [in] condition
  193. * @param [in] value to compare
  194. * @param [in] true_label goto label
  195. * @param [in] stream to submit label_switch task
  196. * @return RT_ERROR_NONE for ok
  197. * @return RT_ERROR_INVALID_VALUE for error input
  198. */
  199. RTS_API rtError_t rtLabelSwitch(void *ptr, rtCondition_t condition, uint32_t value, rtLabel_t trueLabel,
  200. rtStream_t stream);
  201. /**
  202. * @ingroup dvrt_base
  203. * @brief goto label instance
  204. * @param [in] label goto label
  205. * @param [in] stream to submit label_goto task
  206. * @return RT_ERROR_NONE for ok
  207. * @return RT_ERROR_INVALID_VALUE for error input
  208. */
  209. RTS_API rtError_t rtLabelGoto(rtLabel_t label, rtStream_t stream);
  210. /**
  211. * @ingroup dvrt_base
  212. * @brief name label instance
  213. * @param [in] label instance
  214. * @param [in] name label name
  215. * @return RT_ERROR_NONE for ok
  216. * @return RT_ERROR_INVALID_VALUE for error input
  217. */
  218. RTS_API rtError_t rtNameLabel(rtLabel_t label, const char *name);
  219. /**
  220. * @ingroup dvrt_base
  221. * @brief label switch by index
  222. * @param [in] ptr index value ptr
  223. * @param [in] max index max value
  224. * @param [in] labelInfoPtr label content info ptr
  225. * @param [in] stream set stream
  226. * @return RT_ERROR_NONE for ok
  227. * @return RT_ERROR_INVALID_VALUE for error input
  228. */
  229. RTS_API rtError_t rtLabelSwitchByIndex(void *ptr, uint32_t max, void *labelInfoPtr, rtStream_t stream);
  230. /**
  231. * @ingroup dvrt_base
  232. * @brief stream goto label
  233. * @param [in] label goto label
  234. * @param [in] stream stream to submit label_goto task
  235. * @return RT_ERROR_NONE for ok
  236. * @return RT_ERROR_INVALID_VALUE for error input
  237. */
  238. RTS_API rtError_t rtLabelGotoEx(rtLabel_t label, rtStream_t stream);
  239. /**
  240. * @ingroup dvrt_base
  241. * @brief labels to dev info
  242. * @param [in] label model label list
  243. * @param [in] labelNumber label number
  244. * @param [in] dst device ptr
  245. * @param [in] dstMax dst size
  246. * @return RT_ERROR_NONE for ok
  247. * @return RT_ERROR_INVALID_VALUE for error input
  248. */
  249. RTS_API rtError_t rtLabelListCpy(rtLabel_t *label, uint32_t labelNumber, void *dst, uint32_t dstMax);
  250. /**
  251. * @ingroup dvrt_base
  252. * @brief labels to dev info
  253. * @param [out] label created label handle
  254. * @param [in] stream label bind stream
  255. * @return RT_ERROR_NONE for ok
  256. * @return RT_ERROR_INVALID_VALUE for error input
  257. */
  258. RTS_API rtError_t rtLabelCreateEx(rtLabel_t *label, rtStream_t stream);
  259. /**
  260. * @ingroup dvrt_base
  261. * @brief get current thread last stream id and task id
  262. * @param [out] stream id and task id
  263. * @param [in] null
  264. * @return RT_ERROR_NONE for ok
  265. * @return RT_ERROR_INVALID_VALUE for input null ptr
  266. */
  267. RTS_API rtError_t rtGetTaskIdAndStreamID(uint32_t *taskId, uint32_t *streamId);
  268. #if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
  269. }
  270. #endif
  271. #endif // __CCE_RUNTIME_BASE_H__

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