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.

tdt_host_interface.h 6.5 kB

5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  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 HOST_INNER_INC_TDT_HOST_INTERFACE_H_
  17. #define HOST_INNER_INC_TDT_HOST_INTERFACE_H_
  18. #include <string.h>
  19. #include <memory>
  20. #include <vector>
  21. #include "tdt/data_common.h"
  22. #ifdef __cplusplus
  23. extern "C" {
  24. #endif // __cplusplus
  25. namespace tdt {
  26. /**
  27. * @ingroup TdtHostInit
  28. * @brief Initialize the interface, start and initialize various general thread, log and other services
  29. *
  30. * @par Function
  31. * Initialize the interface, start and initialize various general thread, log and other services
  32. *
  33. * @param deviceId [IN] type #unsigned int. Physical device ID
  34. * @retval #0 Success
  35. * @retval #Not 0 Fail
  36. *
  37. * @par Dependency
  38. * @li libtsdclient.so: Library to which the interface belongs.
  39. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  40. */
  41. int32_t TdtHostInit(uint32_t deviceId);
  42. /**
  43. * @ingroup TdtHostPushData
  44. * @brief Blocking queue. When the queue is full, the Push interface will block.
  45. *
  46. * @par Function
  47. * Blocking queue. When the queue is full, the Push interface will block.
  48. *
  49. * @param channelName [IN] type #String. queue channel name
  50. * @param items [IN] type #vector<DataItem> DataItem is defined in data_common.h. input data
  51. * @retval 0 Success
  52. * @retval OtherValues 0 Fail
  53. *
  54. * @par Dependency
  55. * @li libtsdclient.so: Library to which the interface belongs.
  56. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  57. * @li data_common.h: Header file where 'DataItem' defined
  58. */
  59. int32_t TdtHostPushData(const std::string &channelName, const std::vector<DataItem> &item, uint32_t deviceId = 0);
  60. /**
  61. * @ingroup TdtHostDestroy
  62. * @brief Notify TDT component to close related resources
  63. *
  64. * @par Function
  65. * Notify TDT component to close related resources
  66. *
  67. * @param NA
  68. * @retval 0 Success
  69. * @retval OtherValues Fail
  70. *
  71. * @par Dependency
  72. * @li libtsdclient.so: Library to which the interface belongs.
  73. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  74. */
  75. int32_t TdtHostDestroy();
  76. /**
  77. * @ingroup TdtHostPreparePopData
  78. * @brief Prepare pop data from Tdt data storage queue
  79. *
  80. * @par Function
  81. * Prepare pop data from Tdt data storage queue
  82. *
  83. * @param NA
  84. * @retval 0 Success
  85. * @retval OtherValues 0 Fail
  86. *
  87. * @par Dependency
  88. * @li libtsdclient.so: Library to which the interface belongs.
  89. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  90. * @li data_common.h: Header file where 'DataItem' defined
  91. */
  92. int32_t TdtHostPreparePopData();
  93. /**
  94. * @ingroup TdtHostPopData
  95. * @brief POP data from Tdt data storage queue
  96. *
  97. * @par Function
  98. * POP data from Tdt data storage queue
  99. *
  100. * @param channelName [IN] type #String. queue channel name
  101. * @param items [IN] type #vector<DataItem> DataItem is defined in data_common.h. input data
  102. * @retval 0 Success
  103. * @retval OtherValues 0 Fail
  104. *
  105. * @par Dependency
  106. * @li libtsdclient.so: Library to which the interface belongs.
  107. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  108. * @li data_common.h: Header file where 'DataItem' defined
  109. */
  110. int32_t TdtHostPopData(const std::string &channelName, std::vector<DataItem> &item);
  111. /**
  112. * @ingroup TdtHostStop
  113. * @brief Activate the thread that reads data externally from Tdt and
  114. * send end of sequence data so that the external thread can exit
  115. *
  116. * @par Function
  117. * Activate the thread that reads data externally from Tdt and send
  118. * end of sequence data so that the external thread can exit
  119. *
  120. * @param channelName [IN] type #String. queue channel name
  121. * @retval 0 Success
  122. * @retval OtherValues Fail
  123. *
  124. * @par Dependency
  125. * @li libtsdclient.so: Library to which the interface belongs.
  126. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  127. */
  128. int32_t TdtHostStop(const std::string &channelName);
  129. /**
  130. * @ingroup TdtInFeedInit
  131. * @brief Initialize the interface, start and initialize various general thread, log and other services
  132. *
  133. * @par Function
  134. * Initialize the interface, start and initialize various general thread, log and other services
  135. *
  136. * @param deviceId [IN] type #unsigned int. logic device ID
  137. * @retval #0 Success
  138. * @retval #Not 0 Fail
  139. *
  140. * @par Dependency
  141. * @li libtsdclient.so: Library to which the interface belongs.
  142. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  143. */
  144. int32_t TdtInFeedInit(uint32_t deviceId);
  145. /**
  146. * @ingroup TdtOutFeedInit
  147. * @brief Initialize the interface, start and initialize various general thread, log and other services
  148. *
  149. * @par Function
  150. * Initialize the interface, start and initialize various general thread, log and other services
  151. *
  152. * @param deviceId [IN] type #unsigned int. logic device ID
  153. * @retval #0 Success
  154. * @retval #Not 0 Fail
  155. *
  156. * @par Dependency
  157. * @li libtsdclient.so: Library to which the interface belongs.
  158. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  159. */
  160. int32_t TdtOutFeedInit(uint32_t deviceId);
  161. /**
  162. * @ingroup TdtInFeedDestroy
  163. * @brief Notify TDT component to close related resources
  164. *
  165. * @par Function
  166. * Notify TDT component to close related resources
  167. *
  168. * @param NA
  169. * @retval 0 Success
  170. * @retval OtherValues Fail
  171. *
  172. * @par Dependency
  173. * @li libtsdclient.so: Library to which the interface belongs.
  174. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  175. */
  176. int32_t TdtInFeedDestroy(uint32_t deviceId);
  177. /**
  178. * @ingroup TdtOutFeedDestroy
  179. * @brief Notify TDT component to close related resources
  180. *
  181. * @par Function
  182. * Notify TDT component to close related resources
  183. *
  184. * @param NA
  185. * @retval 0 Success
  186. * @retval OtherValues Fail
  187. *
  188. * @par Dependency
  189. * @li libtsdclient.so: Library to which the interface belongs.
  190. * @li tdt_host_interface.h: Header file where the interface declaration is located.
  191. */
  192. int32_t TdtOutFeedDestroy();
  193. } // namespace tdt
  194. #ifdef __cplusplus
  195. }
  196. #endif // __cplusplus
  197. #endif // HOST_INNER_INC_TDT_HOST_INTERFACE_H_

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