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.

acl_tdt_queue.h 12 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426
  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_ACL_ACL_TDT_QUEUE_H_
  17. #define INC_EXTERNAL_ACL_ACL_TDT_QUEUE_H_
  18. #include "acl/acl_base.h"
  19. #ifdef __cplusplus
  20. extern "C" {
  21. #endif
  22. #define ACL_TDT_QUEUE_PERMISSION_MANAGE 1
  23. #define ACL_TDT_QUEUE_PERMISSION_DEQUEUE 2
  24. #define ACL_TDT_QUEUE_PERMISSION_ENQUEUE 4
  25. typedef void *acltdtBuf;
  26. typedef struct tagMemQueueAttr acltdtQueueAttr;
  27. typedef struct acltdtQueueRouteList acltdtQueueRouteList;
  28. typedef struct acltdtQueueRouteQueryInfo acltdtQueueRouteQueryInfo;
  29. typedef struct acltdtQueueRoute acltdtQueueRoute;
  30. typedef enum { ACL_TDT_QUEUE_NAME_PTR = 0, ACL_TDT_QUEUE_DEPTH_UINT32 } acltdtQueueAttrType;
  31. typedef enum {
  32. ACL_TDT_QUEUE_ROUTE_SRC_UINT32 = 0,
  33. ACL_TDT_QUEUE_ROUTE_DST_UINT32,
  34. ACL_TDT_QUEUE_ROUTE_STATUS_INT32
  35. } acltdtQueueRouteParamType;
  36. typedef enum {
  37. ACL_TDT_QUEUE_ROUTE_QUERY_SRC = 0,
  38. ACL_TDT_QUEUE_ROUTE_QUERY_DST,
  39. ACL_TDT_QUEUE_ROUTE_QUERY_SRC_AND_DST
  40. } acltdtQueueRouteQueryMode;
  41. typedef enum {
  42. ACL_TDT_QUEUE_ROUTE_QUERY_MODE_ENUM = 0,
  43. ACL_TDT_QUEUE_ROUTE_QUERY_SRC_ID_UINT32,
  44. ACL_TDT_QUEUE_ROUTE_QUERY_DST_ID_UINT32
  45. } acltdtQueueRouteQueryInfoParamType;
  46. /**
  47. * @ingroup AscendCL
  48. * @brief create queue
  49. *
  50. * @param attr [IN] pointer to the queue attr
  51. * @param qid [OUT] pointer to the qid
  52. *
  53. * @retval ACL_SUCCESS The function is successfully executed.
  54. * @retval OtherValues Failure
  55. *
  56. * @see acltdtDestroyQueue
  57. */
  58. ACL_FUNC_VISIBILITY aclError acltdtCreateQueue(const acltdtQueueAttr *attr, uint32_t *qid);
  59. /**
  60. * @ingroup AscendCL
  61. * @brief destroy queue
  62. *
  63. * @param qid [IN] qid which to be destroyed
  64. *
  65. * @retval ACL_SUCCESS The function is successfully executed.
  66. * @retval OtherValues Failure
  67. *
  68. * @see acltdtCreateQueue
  69. */
  70. ACL_FUNC_VISIBILITY aclError acltdtDestroyQueue(uint32_t qid);
  71. /**
  72. * @ingroup AscendCL
  73. * @brief enqueue function
  74. *
  75. * @param qid [IN] qid
  76. * @param buf [IN] acltdtBuf
  77. * @param timeout [IN] timeout
  78. *
  79. * @retval ACL_SUCCESS The function is successfully executed.
  80. * @retval OtherValues Failure
  81. *
  82. * @see acltdtDequeue
  83. */
  84. ACL_FUNC_VISIBILITY aclError acltdtEnqueue(uint32_t qid, acltdtBuf buf, int32_t timeout);
  85. /**
  86. * @ingroup AscendCL
  87. * @brief dequeue function
  88. *
  89. * @param qid [IN] qid
  90. * @param buf [OUT] pointer to the acltdtBuf
  91. * @param timeout [IN] timeout
  92. *
  93. * @retval ACL_SUCCESS The function is successfully executed.
  94. * @retval OtherValues Failure
  95. *
  96. * @see acltdtEnqueue
  97. */
  98. ACL_FUNC_VISIBILITY aclError acltdtDequeue(uint32_t qid, acltdtBuf *buf, int32_t timeout);
  99. /**
  100. * @ingroup AscendCL
  101. * @brief grant queue to other process
  102. *
  103. * @param qid [IN] qid
  104. * @param pid [IN] pid of dst process
  105. * @param permission [IN] permission of queue
  106. * @param timeout [IN] timeout
  107. *
  108. * @retval ACL_SUCCESS The function is successfully executed.
  109. * @retval OtherValues Failure
  110. *
  111. * @see ACL_TDT_QUEUE_PERMISSION_MANAGE | ACL_TDT_QUEUE_PERMISSION_DEQUEUE | ACL_TDT_QUEUE_PERMISSION_ENQUEUE
  112. */
  113. ACL_FUNC_VISIBILITY aclError acltdtGrantQueue(uint32_t qid, int32_t pid, uint32_t permission, int32_t timeout);
  114. /**
  115. * @ingroup AscendCL
  116. * @brief attach queue in current process
  117. *
  118. * @param qid [IN] qid
  119. * @param timeout [IN] timeout
  120. * @param permission [OUT] permission of queue
  121. *
  122. * @retval ACL_SUCCESS The function is successfully executed.
  123. * @retval OtherValues Failure
  124. *
  125. * @see acltdtGrantQueue
  126. */
  127. ACL_FUNC_VISIBILITY aclError acltdtAttachQueue(uint32_t qid, int32_t timeout, uint32_t *permission);
  128. /**
  129. * @ingroup AscendCL
  130. * @brief bind queue routes
  131. *
  132. * @param qRouteList [IN|OUT] pointer to the route list
  133. *
  134. * @retval ACL_SUCCESS The function is successfully executed.
  135. * @retval OtherValues Failure
  136. */
  137. ACL_FUNC_VISIBILITY aclError acltdtBindQueueRoutes(acltdtQueueRouteList *qRouteList);
  138. /**
  139. * @ingroup AscendCL
  140. * @brief unbind queue routes
  141. *
  142. * @param qRouteList [IN|OUT] pointer to the route list
  143. *
  144. * @retval ACL_SUCCESS The function is successfully executed.
  145. * @retval OtherValues Failure
  146. */
  147. ACL_FUNC_VISIBILITY aclError acltdtUnbindQueueRoutes(acltdtQueueRouteList *qRouteList);
  148. /**
  149. * @ingroup AscendCL
  150. * @brief query queue routes according to query mode
  151. *
  152. * @param queryInfo [IN] pointer to the queue route query info
  153. * @param qRouteList [IN|OUT] pointer to the route list
  154. *
  155. * @retval ACL_SUCCESS The function is successfully executed.
  156. * @retval OtherValues Failure
  157. */
  158. ACL_FUNC_VISIBILITY aclError acltdtQueryQueueRoutes(const acltdtQueueRouteQueryInfo *queryInfo,
  159. acltdtQueueRouteList *qRouteList);
  160. /**
  161. * @ingroup AscendCL
  162. * @brief alloc acltdtBuf
  163. *
  164. * @param size [IN] alloc buf size
  165. * @param buf [OUT] pointer to the acltdtBuf
  166. *
  167. * @retval ACL_SUCCESS The function is successfully executed.
  168. * @retval OtherValues Failure
  169. *
  170. * @see acltdtFreeBuf
  171. */
  172. ACL_FUNC_VISIBILITY aclError acltdtAllocBuf(size_t size, acltdtBuf *buf);
  173. /**
  174. * @ingroup AscendCL
  175. * @brief free acltdtBuf
  176. *
  177. * @param buf [IN] pointer to the acltdtBuf
  178. *
  179. * @retval ACL_SUCCESS The function is successfully executed.
  180. * @retval OtherValues Failure
  181. *
  182. * @see acltdtAllocBuf
  183. */
  184. ACL_FUNC_VISIBILITY aclError acltdtFreeBuf(acltdtBuf buf);
  185. /**
  186. * @ingroup AscendCL
  187. * @brief get data buf address
  188. *
  189. * @param buf [IN] acltdtBuf
  190. * @param dataPtr [OUT] pointer to the data ptr which is acquired from acltdtBuf
  191. * @param size [OUT] pointer to the size
  192. *
  193. * @retval ACL_SUCCESS The function is successfully executed.
  194. * @retval OtherValues Failure
  195. *
  196. * @see acltdtAllocBuf
  197. */
  198. ACL_FUNC_VISIBILITY aclError acltdtGetBufData(const acltdtBuf buf, void **dataPtr, size_t *size);
  199. /**
  200. * @ingroup AscendCL
  201. * @brief Create the queue attr
  202. *
  203. * @retval null for failed
  204. * @retval OtherValues success
  205. *
  206. * @see acltdtDestroyQueueAttr
  207. */
  208. ACL_FUNC_VISIBILITY acltdtQueueAttr *acltdtCreateQueueAttr();
  209. /**
  210. * @ingroup AscendCL
  211. * @brief Destroy the queue attr
  212. *
  213. * @param attr [IN] pointer to the queue attr
  214. *
  215. * @retval ACL_SUCCESS The function is successfully executed.
  216. * @retval OtherValues Failure
  217. *
  218. * @see acltdtCreateQueueAttr
  219. */
  220. ACL_FUNC_VISIBILITY aclError acltdtDestroyQueueAttr(const acltdtQueueAttr *attr);
  221. /**
  222. * @ingroup AscendCL
  223. * @brief Set parameter for queue attr
  224. *
  225. * @param attr [IN|OUT] pointer to the queue attr
  226. * @param type [IN] parameter type
  227. * @param len [IN] parameter length
  228. * @param param [IN] pointer to parameter value
  229. *
  230. * @retval ACL_SUCCESS for success, other for failure
  231. *
  232. * @see acltdtCreateQueueAttr
  233. */
  234. ACL_FUNC_VISIBILITY aclError acltdtSetQueueAttr(acltdtQueueAttr *attr, acltdtQueueAttrType type, size_t len,
  235. const void *param);
  236. /**
  237. * @ingroup AscendCL
  238. *
  239. * @brief Get parameter for queue attr.
  240. *
  241. * @param attr [IN] pointer to the queue attr
  242. * @param type [IN] parameter type
  243. * @param len [IN] parameter length
  244. * @param paramRetSize [OUT] pointer to parameter real length
  245. * @param param [OUT] pointer to parameter value
  246. *
  247. * @retval ACL_SUCCESS for success, other for failure
  248. *
  249. * @see acltdtCreateQueueAttr
  250. */
  251. ACL_FUNC_VISIBILITY aclError acltdtGetQueueAttr(const acltdtQueueAttr *attr, acltdtQueueAttrType type, size_t len,
  252. size_t *paramRetSize, void *param);
  253. /**
  254. * @ingroup AscendCL
  255. * @brief Create the queue route
  256. *
  257. * @param srcId [IN] src id of queue route
  258. * @param dstId [IN] dst id of queue route
  259. *
  260. * @retval null for failed
  261. * @retval OtherValues success
  262. *
  263. * @see acltdtDestroyQueueRoute
  264. */
  265. ACL_FUNC_VISIBILITY acltdtQueueRoute *acltdtCreateQueueRoute(uint32_t srcId, uint32_t dstId);
  266. /**
  267. * @ingroup AscendCL
  268. * @brief Destroy the queue attr
  269. *
  270. * @param route [IN] pointer to the queue route
  271. *
  272. * @retval ACL_SUCCESS The function is successfully executed.
  273. * @retval OtherValues Failure
  274. *
  275. * @see acltdtCreateQueueRoute
  276. */
  277. ACL_FUNC_VISIBILITY aclError acltdtDestroyQueueRoute(const acltdtQueueRoute *route);
  278. /**
  279. * @ingroup AscendCL
  280. *
  281. * @brief Get parameter for queue route.
  282. *
  283. * @param route [IN] pointer to the queue route
  284. * @param type [IN] parameter type
  285. * @param len [IN] parameter length
  286. * @param paramRetSize [OUT] pointer to parameter real length
  287. * @param param [OUT] pointer to parameter value
  288. *
  289. * @retval ACL_SUCCESS for success, other for failure
  290. *
  291. * @see acltdtCreateQueueRoute
  292. */
  293. ACL_FUNC_VISIBILITY aclError acltdtGetQueueRouteParam(const acltdtQueueRoute *route, acltdtQueueRouteParamType type,
  294. size_t len, size_t *paramRetSize, void *param);
  295. /**
  296. * @ingroup AscendCL
  297. * @brief Create the queue route list
  298. *
  299. * @retval null for failed
  300. * @retval OtherValues success
  301. *
  302. * @see acltdtDestroyQueueRouteList
  303. */
  304. ACL_FUNC_VISIBILITY acltdtQueueRouteList *acltdtCreateQueueRouteList();
  305. /**
  306. * @ingroup AscendCL
  307. * @brief Destroy the queue route list
  308. *
  309. * @param routeList [IN] pointer to the queue route list
  310. *
  311. * @retval ACL_SUCCESS The function is successfully executed.
  312. * @retval OtherValues Failure
  313. *
  314. * @see acltdtCreateQueueRouteList
  315. */
  316. ACL_FUNC_VISIBILITY aclError acltdtDestroyQueueRouteList(const acltdtQueueRouteList *routeList);
  317. /**
  318. * @ingroup AscendCL
  319. * @brief add queue route to the route list
  320. *
  321. * @param routeList [IN|OUT] pointer to the queue route list
  322. * @param route [IN] pointer to the queue route
  323. *
  324. * @retval ACL_SUCCESS The function is successfully executed.
  325. * @retval OtherValues Failure
  326. *
  327. * @see acltdtCreateQueueRouteList | acltdtCreateQueueRoute
  328. *
  329. */
  330. ACL_FUNC_VISIBILITY aclError acltdtAddQueueRoute(acltdtQueueRouteList *routeList, const acltdtQueueRoute *route);
  331. /**
  332. * @ingroup AscendCL
  333. * @brief get queue route from route list
  334. *
  335. * @param routeList [IN] pointer to the queue route list
  336. * @param index [IN] index of queue route in route list
  337. * @param route [IN|OUT] pointer to the queue route
  338. *
  339. * @retval ACL_SUCCESS The function is successfully executed.
  340. * @retval OtherValues Failure
  341. *
  342. * @see acltdtCreateQueueRouteList | acltdtCreateQueueRoute
  343. *
  344. */
  345. ACL_FUNC_VISIBILITY aclError acltdtGetQueueRoute(const acltdtQueueRouteList *routeList, size_t index,
  346. acltdtQueueRoute *route);
  347. /**
  348. * @ingroup AscendCL
  349. * @brief Create the queue route query info
  350. *
  351. * @retval null for failed
  352. * @retval OtherValues success
  353. *
  354. * @see acltdtDestroyQueueRouteQueryInfo
  355. */
  356. ACL_FUNC_VISIBILITY acltdtQueueRouteQueryInfo *acltdtCreateQueueRouteQueryInfo();
  357. /**
  358. * @ingroup AscendCL
  359. * @brief Destroy the queue route query info
  360. *
  361. * @param info [IN] pointer to the queue route info
  362. *
  363. * @retval ACL_SUCCESS The function is successfully executed.
  364. * @retval OtherValues Failure
  365. *
  366. * @see acltdtCreateQueueRouteQueryInfo
  367. *
  368. */
  369. ACL_FUNC_VISIBILITY aclError acltdtDestroyQueueRouteQueryInfo(const acltdtQueueRouteQueryInfo *info);
  370. /**
  371. * @ingroup AscendCL
  372. * @brief Set parameter for queue route info
  373. *
  374. * @param attr [IN|OUT] pointer to the queue route info
  375. * @param type [IN] parameter type
  376. * @param len [IN] parameter length
  377. * @param param [IN] pointer to parameter value
  378. *
  379. * @retval ACL_SUCCESS for success, other for failure
  380. *
  381. * @see acltdtCreateQueueRouteQueryInfo
  382. */
  383. ACL_FUNC_VISIBILITY aclError acltdtSetQueueRouteQueryInfo(acltdtQueueRouteQueryInfo *param,
  384. acltdtQueueRouteQueryInfoParamType type, size_t len,
  385. const void *value);
  386. #ifdef __cplusplus
  387. }
  388. #endif
  389. #endif // INC_EXTERNAL_ACL_ACL_TDT_QUEUE_H_

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