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_base.h 23 kB

3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
3 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656
  1. /**
  2. * Copyright 2019-2022 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_BASE_H_
  17. #define INC_EXTERNAL_ACL_ACL_BASE_H_
  18. #include <stdint.h>
  19. #include <stddef.h>
  20. #include "error_codes/rt_error_codes.h"
  21. #include "error_codes/ge_error_codes.h"
  22. #ifdef __cplusplus
  23. extern "C" {
  24. #endif
  25. #if defined(_MSC_VER)
  26. #ifdef FUNC_VISIBILITY
  27. #define ACL_FUNC_VISIBILITY _declspec(dllexport)
  28. #else
  29. #define ACL_FUNC_VISIBILITY
  30. #endif
  31. #else
  32. #ifdef FUNC_VISIBILITY
  33. #define ACL_FUNC_VISIBILITY __attribute__((visibility("default")))
  34. #else
  35. #define ACL_FUNC_VISIBILITY
  36. #endif
  37. #endif
  38. #ifdef __GNUC__
  39. #define ACL_DEPRECATED __attribute__((deprecated))
  40. #define ACL_DEPRECATED_MESSAGE(message) __attribute__((deprecated(message)))
  41. #elif defined(_MSC_VER)
  42. #define ACL_DEPRECATED __declspec(deprecated)
  43. #define ACL_DEPRECATED_MESSAGE(message) __declspec(deprecated(message))
  44. #else
  45. #define ACL_DEPRECATED
  46. #define ACL_DEPRECATED_MESSAGE(message)
  47. #endif
  48. typedef void *aclrtStream;
  49. typedef void *aclrtEvent;
  50. typedef void *aclrtContext;
  51. typedef int aclError;
  52. typedef uint16_t aclFloat16;
  53. typedef struct aclDataBuffer aclDataBuffer;
  54. typedef struct aclTensorDesc aclTensorDesc;
  55. static const int ACL_ERROR_NONE = 0;
  56. static const int ACL_SUCCESS = 0;
  57. static const int ACL_ERROR_INVALID_PARAM = 100000;
  58. static const int ACL_ERROR_UNINITIALIZE = 100001;
  59. static const int ACL_ERROR_REPEAT_INITIALIZE = 100002;
  60. static const int ACL_ERROR_INVALID_FILE = 100003;
  61. static const int ACL_ERROR_WRITE_FILE = 100004;
  62. static const int ACL_ERROR_INVALID_FILE_SIZE = 100005;
  63. static const int ACL_ERROR_PARSE_FILE = 100006;
  64. static const int ACL_ERROR_FILE_MISSING_ATTR = 100007;
  65. static const int ACL_ERROR_FILE_ATTR_INVALID = 100008;
  66. static const int ACL_ERROR_INVALID_DUMP_CONFIG = 100009;
  67. static const int ACL_ERROR_INVALID_PROFILING_CONFIG = 100010;
  68. static const int ACL_ERROR_INVALID_MODEL_ID = 100011;
  69. static const int ACL_ERROR_DESERIALIZE_MODEL = 100012;
  70. static const int ACL_ERROR_PARSE_MODEL = 100013;
  71. static const int ACL_ERROR_READ_MODEL_FAILURE = 100014;
  72. static const int ACL_ERROR_MODEL_SIZE_INVALID = 100015;
  73. static const int ACL_ERROR_MODEL_MISSING_ATTR = 100016;
  74. static const int ACL_ERROR_MODEL_INPUT_NOT_MATCH = 100017;
  75. static const int ACL_ERROR_MODEL_OUTPUT_NOT_MATCH = 100018;
  76. static const int ACL_ERROR_MODEL_NOT_DYNAMIC = 100019;
  77. static const int ACL_ERROR_OP_TYPE_NOT_MATCH = 100020;
  78. static const int ACL_ERROR_OP_INPUT_NOT_MATCH = 100021;
  79. static const int ACL_ERROR_OP_OUTPUT_NOT_MATCH = 100022;
  80. static const int ACL_ERROR_OP_ATTR_NOT_MATCH = 100023;
  81. static const int ACL_ERROR_OP_NOT_FOUND = 100024;
  82. static const int ACL_ERROR_OP_LOAD_FAILED = 100025;
  83. static const int ACL_ERROR_UNSUPPORTED_DATA_TYPE = 100026;
  84. static const int ACL_ERROR_FORMAT_NOT_MATCH = 100027;
  85. static const int ACL_ERROR_BIN_SELECTOR_NOT_REGISTERED = 100028;
  86. static const int ACL_ERROR_KERNEL_NOT_FOUND = 100029;
  87. static const int ACL_ERROR_BIN_SELECTOR_ALREADY_REGISTERED = 100030;
  88. static const int ACL_ERROR_KERNEL_ALREADY_REGISTERED = 100031;
  89. static const int ACL_ERROR_INVALID_QUEUE_ID = 100032;
  90. static const int ACL_ERROR_REPEAT_SUBSCRIBE = 100033;
  91. static const int ACL_ERROR_STREAM_NOT_SUBSCRIBE = 100034;
  92. static const int ACL_ERROR_THREAD_NOT_SUBSCRIBE = 100035;
  93. static const int ACL_ERROR_WAIT_CALLBACK_TIMEOUT = 100036;
  94. static const int ACL_ERROR_REPEAT_FINALIZE = 100037;
  95. static const int ACL_ERROR_NOT_STATIC_AIPP = 100038;
  96. static const int ACL_ERROR_COMPILING_STUB_MODE = 100039;
  97. static const int ACL_ERROR_GROUP_NOT_SET = 100040;
  98. static const int ACL_ERROR_GROUP_NOT_CREATE = 100041;
  99. static const int ACL_ERROR_PROF_ALREADY_RUN = 100042;
  100. static const int ACL_ERROR_PROF_NOT_RUN = 100043;
  101. static const int ACL_ERROR_DUMP_ALREADY_RUN = 100044;
  102. static const int ACL_ERROR_DUMP_NOT_RUN = 100045;
  103. static const int ACL_ERROR_PROF_REPEAT_SUBSCRIBE = 148046;
  104. static const int ACL_ERROR_PROF_API_CONFLICT = 148047;
  105. static const int ACL_ERROR_INVALID_MAX_OPQUEUE_NUM_CONFIG = 148048;
  106. static const int ACL_ERROR_INVALID_OPP_PATH = 148049;
  107. static const int ACL_ERROR_OP_UNSUPPORTED_DYNAMIC = 148050;
  108. static const int ACL_ERROR_RELATIVE_RESOURCE_NOT_CLEARED = 148051;
  109. static const int ACL_ERROR_UNSUPPORTED_JPEG = 148052;
  110. static const int ACL_ERROR_BAD_ALLOC = 200000;
  111. static const int ACL_ERROR_API_NOT_SUPPORT = 200001;
  112. static const int ACL_ERROR_INVALID_DEVICE = 200002;
  113. static const int ACL_ERROR_MEMORY_ADDRESS_UNALIGNED = 200003;
  114. static const int ACL_ERROR_RESOURCE_NOT_MATCH = 200004;
  115. static const int ACL_ERROR_INVALID_RESOURCE_HANDLE = 200005;
  116. static const int ACL_ERROR_FEATURE_UNSUPPORTED = 200006;
  117. static const int ACL_ERROR_PROF_MODULES_UNSUPPORTED = 200007;
  118. static const int ACL_ERROR_STORAGE_OVER_LIMIT = 300000;
  119. static const int ACL_ERROR_INTERNAL_ERROR = 500000;
  120. static const int ACL_ERROR_FAILURE = 500001;
  121. static const int ACL_ERROR_GE_FAILURE = 500002;
  122. static const int ACL_ERROR_RT_FAILURE = 500003;
  123. static const int ACL_ERROR_DRV_FAILURE = 500004;
  124. static const int ACL_ERROR_PROFILING_FAILURE = 500005;
  125. #define ACL_TENSOR_SHAPE_RANGE_NUM 2
  126. #define ACL_TENSOR_VALUE_RANGE_NUM 2
  127. #define ACL_UNKNOWN_RANK 0xFFFFFFFFFFFFFFFE
  128. typedef enum {
  129. ACL_DT_UNDEFINED = -1,
  130. ACL_FLOAT = 0,
  131. ACL_FLOAT16 = 1,
  132. ACL_INT8 = 2,
  133. ACL_INT32 = 3,
  134. ACL_UINT8 = 4,
  135. ACL_INT16 = 6,
  136. ACL_UINT16 = 7,
  137. ACL_UINT32 = 8,
  138. ACL_INT64 = 9,
  139. ACL_UINT64 = 10,
  140. ACL_DOUBLE = 11,
  141. ACL_BOOL = 12,
  142. ACL_STRING = 13,
  143. ACL_COMPLEX64 = 16,
  144. ACL_COMPLEX128 = 17,
  145. ACL_BF16 = 27
  146. } aclDataType;
  147. typedef enum {
  148. ACL_FORMAT_UNDEFINED = -1,
  149. ACL_FORMAT_NCHW = 0,
  150. ACL_FORMAT_NHWC = 1,
  151. ACL_FORMAT_ND = 2,
  152. ACL_FORMAT_NC1HWC0 = 3,
  153. ACL_FORMAT_FRACTAL_Z = 4,
  154. ACL_FORMAT_NC1HWC0_C04 = 12,
  155. ACL_FORMAT_HWCN = 16,
  156. ACL_FORMAT_NDHWC = 27,
  157. ACL_FORMAT_FRACTAL_NZ = 29,
  158. ACL_FORMAT_NCDHW = 30,
  159. ACL_FORMAT_NDC1HWC0 = 32,
  160. ACL_FRACTAL_Z_3D = 33
  161. } aclFormat;
  162. typedef enum {
  163. ACL_DEBUG = 0,
  164. ACL_INFO = 1,
  165. ACL_WARNING = 2,
  166. ACL_ERROR = 3,
  167. } aclLogLevel;
  168. typedef enum { ACL_MEMTYPE_DEVICE = 0, ACL_MEMTYPE_HOST = 1, ACL_MEMTYPE_HOST_COMPILE_INDEPENDENT = 2 } aclMemType;
  169. /**
  170. * @ingroup AscendCL
  171. * @brief Converts data of type aclFloat16 to data of type float
  172. *
  173. * @param value [IN] Data to be converted
  174. *
  175. * @retval Transformed data
  176. */
  177. ACL_FUNC_VISIBILITY float aclFloat16ToFloat(aclFloat16 value);
  178. /**
  179. * @ingroup AscendCL
  180. * @brief Converts data of type float to data of type aclFloat16
  181. *
  182. * @param value [IN] Data to be converted
  183. *
  184. * @retval Transformed data
  185. */
  186. ACL_FUNC_VISIBILITY aclFloat16 aclFloatToFloat16(float value);
  187. /**
  188. * @ingroup AscendCL
  189. * @brief create data of aclDataBuffer
  190. *
  191. * @param data [IN] pointer to data
  192. * @li Need to be managed by the user,
  193. * call aclrtMalloc interface to apply for memory,
  194. * call aclrtFree interface to release memory
  195. *
  196. * @param size [IN] size of data in bytes
  197. *
  198. * @retval pointer to created instance. nullptr if run out of memory
  199. *
  200. * @see aclrtMalloc | aclrtFree
  201. */
  202. ACL_FUNC_VISIBILITY aclDataBuffer *aclCreateDataBuffer(void *data, size_t size);
  203. /**
  204. * @ingroup AscendCL
  205. * @brief destroy data of aclDataBuffer
  206. *
  207. * @par Function
  208. * Only the aclDataBuffer type data is destroyed here.
  209. * The memory of the data passed in when the aclDataDataBuffer interface
  210. * is called to create aclDataBuffer type data must be released by the user
  211. *
  212. * @param dataBuffer [IN] pointer to the aclDataBuffer
  213. *
  214. * @retval ACL_SUCCESS The function is successfully executed.
  215. * @retval OtherValues Failure
  216. *
  217. * @see aclCreateDataBuffer
  218. */
  219. ACL_FUNC_VISIBILITY aclError aclDestroyDataBuffer(const aclDataBuffer *dataBuffer);
  220. /**
  221. * @ingroup AscendCL
  222. * @brief update new data of aclDataBuffer
  223. *
  224. * @param dataBuffer [OUT] pointer to aclDataBuffer
  225. * @li The old data need to be released by the user, otherwise it may occur memory leak leakage
  226. * call aclGetDataBufferAddr interface to get old data address
  227. * call aclrtFree interface to release memory
  228. *
  229. * @param data [IN] pointer to new data
  230. * @li Need to be managed by the user,
  231. * call aclrtMalloc interface to apply for memory,
  232. * call aclrtFree interface to release memory
  233. *
  234. * @param size [IN] size of data in bytes
  235. *
  236. * @retval ACL_SUCCESS The function is successfully executed.
  237. * @retval OtherValues Failure
  238. *
  239. * @see aclrtMalloc | aclrtFree | aclGetDataBufferAddr
  240. */
  241. ACL_FUNC_VISIBILITY aclError aclUpdateDataBuffer(aclDataBuffer *dataBuffer, void *data, size_t size);
  242. /**
  243. * @ingroup AscendCL
  244. * @brief get data address from aclDataBuffer
  245. *
  246. * @param dataBuffer [IN] pointer to the data of aclDataBuffer
  247. *
  248. * @retval data address
  249. */
  250. ACL_FUNC_VISIBILITY void *aclGetDataBufferAddr(const aclDataBuffer *dataBuffer);
  251. /**
  252. * @ingroup AscendCL
  253. * @brief get data size of aclDataBuffer
  254. *
  255. * @param dataBuffer [IN] pointer to the data of aclDataBuffer
  256. *
  257. * @retval data size
  258. */
  259. ACL_DEPRECATED_MESSAGE("aclGetDataBufferSize is deprecated, use aclGetDataBufferSizeV2 instead")
  260. ACL_FUNC_VISIBILITY uint32_t aclGetDataBufferSize(const aclDataBuffer *dataBuffer);
  261. /**
  262. * @ingroup AscendCL
  263. * @brief get data size of aclDataBuffer to replace aclGetDataBufferSize
  264. *
  265. * @param dataBuffer [IN] pointer to the data of aclDataBuffer
  266. *
  267. * @retval data size
  268. */
  269. ACL_FUNC_VISIBILITY size_t aclGetDataBufferSizeV2(const aclDataBuffer *dataBuffer);
  270. /**
  271. * @ingroup AscendCL
  272. * @brief get size of aclDataType
  273. *
  274. * @param dataType [IN] aclDataType data the size to get
  275. *
  276. * @retval size of the aclDataType
  277. */
  278. ACL_FUNC_VISIBILITY size_t aclDataTypeSize(aclDataType dataType);
  279. // interfaces of tensor desc
  280. /**
  281. * @ingroup AscendCL
  282. * @brief create data aclTensorDesc
  283. *
  284. * @param dataType [IN] Data types described by tensor
  285. * @param numDims [IN] the number of dimensions of the shape
  286. * @param dims [IN] the size of the specified dimension
  287. * @param format [IN] tensor format
  288. *
  289. * @retval aclTensorDesc pointer.
  290. * @retval nullptr if param is invalid or run out of memory
  291. */
  292. ACL_FUNC_VISIBILITY aclTensorDesc *aclCreateTensorDesc(aclDataType dataType, int numDims, const int64_t *dims,
  293. aclFormat format);
  294. /**
  295. * @ingroup AscendCL
  296. * @brief destroy data aclTensorDesc
  297. *
  298. * @param desc [IN] pointer to the data of aclTensorDesc to destroy
  299. */
  300. ACL_FUNC_VISIBILITY void aclDestroyTensorDesc(const aclTensorDesc *desc);
  301. /**
  302. * @ingroup AscendCL
  303. * @brief set tensor shape range for aclTensorDesc
  304. *
  305. * @param desc [OUT] pointer to the data of aclTensorDesc
  306. * @param dimsCount [IN] the number of dimensions of the shape
  307. * @param dimsRange [IN] the range of dimensions of the shape
  308. *
  309. * @retval ACL_SUCCESS The function is successfully executed.
  310. * @retval OtherValues Failure
  311. */
  312. ACL_FUNC_VISIBILITY aclError aclSetTensorShapeRange(aclTensorDesc *desc, size_t dimsCount,
  313. int64_t dimsRange[][ACL_TENSOR_SHAPE_RANGE_NUM]);
  314. /**
  315. * @ingroup AscendCL
  316. * @brief set value range for aclTensorDesc
  317. *
  318. * @param desc [OUT] pointer to the data of aclTensorDesc
  319. * @param valueCount [IN] the number of value
  320. * @param valueRange [IN] the range of value
  321. *
  322. * @retval ACL_SUCCESS The function is successfully executed.
  323. * @retval OtherValues Failure
  324. */
  325. ACL_FUNC_VISIBILITY aclError aclSetTensorValueRange(aclTensorDesc *desc, size_t valueCount,
  326. int64_t valueRange[][ACL_TENSOR_VALUE_RANGE_NUM]);
  327. /**
  328. * @ingroup AscendCL
  329. * @brief get data type specified by the tensor description
  330. *
  331. * @param desc [IN] pointer to the instance of aclTensorDesc
  332. *
  333. * @retval data type specified by the tensor description.
  334. * @retval ACL_DT_UNDEFINED if description is null
  335. */
  336. ACL_FUNC_VISIBILITY aclDataType aclGetTensorDescType(const aclTensorDesc *desc);
  337. /**
  338. * @ingroup AscendCL
  339. * @brief get data format specified by the tensor description
  340. *
  341. * @param desc [IN] pointer to the instance of aclTensorDesc
  342. *
  343. * @retval data format specified by the tensor description.
  344. * @retval ACL_FORMAT_UNDEFINED if description is null
  345. */
  346. ACL_FUNC_VISIBILITY aclFormat aclGetTensorDescFormat(const aclTensorDesc *desc);
  347. /**
  348. * @ingroup AscendCL
  349. * @brief get tensor size specified by the tensor description
  350. *
  351. * @param desc [IN] pointer to the instance of aclTensorDesc
  352. *
  353. * @retval data size specified by the tensor description.
  354. * @retval 0 if description is null
  355. */
  356. ACL_FUNC_VISIBILITY size_t aclGetTensorDescSize(const aclTensorDesc *desc);
  357. /**
  358. * @ingroup AscendCL
  359. * @brief get element count specified by the tensor description
  360. *
  361. * @param desc [IN] pointer to the instance of aclTensorDesc
  362. *
  363. * @retval element count specified by the tensor description.
  364. * @retval 0 if description is null
  365. */
  366. ACL_FUNC_VISIBILITY size_t aclGetTensorDescElementCount(const aclTensorDesc *desc);
  367. /**
  368. * @ingroup AscendCL
  369. * @brief get number of dims specified by the tensor description
  370. *
  371. * @param desc [IN] pointer to the instance of aclTensorDesc
  372. *
  373. * @retval number of dims specified by the tensor description.
  374. * @retval 0 if description is null
  375. * @retval ACL_UNKNOWN_RANK if the tensor dim is -2
  376. */
  377. ACL_FUNC_VISIBILITY size_t aclGetTensorDescNumDims(const aclTensorDesc *desc);
  378. /**
  379. * @ingroup AscendCL
  380. * @brief Get the size of the specified dim in the tensor description
  381. *
  382. * @param desc [IN] pointer to the instance of aclTensorDesc
  383. * @param index [IN] index of dims, start from 0.
  384. *
  385. * @retval dim specified by the tensor description and index.
  386. * @retval -1 if description or index is invalid
  387. */
  388. ACL_DEPRECATED_MESSAGE("aclGetTensorDescDim is deprecated, use aclGetTensorDescDimV2 instead")
  389. ACL_FUNC_VISIBILITY int64_t aclGetTensorDescDim(const aclTensorDesc *desc, size_t index);
  390. /**
  391. * @ingroup AscendCL
  392. * @brief Get the size of the specified dim in the tensor description
  393. *
  394. * @param desc [IN] pointer to the instance of aclTensorDesc
  395. * @param index [IN] index of dims, start from 0.
  396. * @param dimSize [OUT] size of the specified dim.
  397. *
  398. * @retval ACL_SUCCESS The function is successfully executed.
  399. * @retval OtherValues Failure
  400. */
  401. ACL_FUNC_VISIBILITY aclError aclGetTensorDescDimV2(const aclTensorDesc *desc, size_t index, int64_t *dimSize);
  402. /**
  403. * @ingroup AscendCL
  404. * @brief Get the range of the specified dim in the tensor description
  405. *
  406. * @param desc [IN] pointer to the instance of aclTensorDesc
  407. * @param index [IN] index of dims, start from 0.
  408. * @param dimRangeNum [IN] number of dimRange.
  409. * @param dimRange [OUT] range of the specified dim.
  410. *
  411. * @retval ACL_SUCCESS The function is successfully executed.
  412. * @retval OtherValues Failure
  413. */
  414. ACL_FUNC_VISIBILITY aclError aclGetTensorDescDimRange(const aclTensorDesc *desc, size_t index, size_t dimRangeNum,
  415. int64_t *dimRange);
  416. /**
  417. * @ingroup AscendCL
  418. * @brief set tensor description name
  419. *
  420. * @param desc [OUT] pointer to the instance of aclTensorDesc
  421. * @param name [IN] tensor description name
  422. */
  423. ACL_FUNC_VISIBILITY void aclSetTensorDescName(aclTensorDesc *desc, const char *name);
  424. /**
  425. * @ingroup AscendCL
  426. * @brief get tensor description name
  427. *
  428. * @param desc [IN] pointer to the instance of aclTensorDesc
  429. *
  430. * @retval tensor description name.
  431. * @retval empty string if description is null
  432. */
  433. ACL_FUNC_VISIBILITY const char *aclGetTensorDescName(aclTensorDesc *desc);
  434. /**
  435. * @ingroup AscendCL
  436. * @brief Convert the format in the source aclTensorDesc according to
  437. * the specified dstFormat to generate a new target aclTensorDesc.
  438. * The format in the source aclTensorDesc remains unchanged.
  439. *
  440. * @param srcDesc [IN] pointer to the source tensor desc
  441. * @param dstFormat [IN] destination format
  442. * @param dstDesc [OUT] pointer to the pointer to the destination tensor desc
  443. *
  444. * @retval ACL_SUCCESS The function is successfully executed.
  445. * @retval OtherValues Failure
  446. */
  447. ACL_FUNC_VISIBILITY aclError aclTransTensorDescFormat(const aclTensorDesc *srcDesc, aclFormat dstFormat,
  448. aclTensorDesc **dstDesc);
  449. /**
  450. * @ingroup AscendCL
  451. * @brief Set the storage format specified by the tensor description
  452. *
  453. * @param desc [OUT] pointer to the instance of aclTensorDesc
  454. * @param format [IN] the storage format
  455. *
  456. * @retval ACL_SUCCESS The function is successfully executed.
  457. * @retval OtherValues Failure
  458. */
  459. ACL_DEPRECATED_MESSAGE("aclSetTensorStorageFormat is deprecated, use aclSetTensorFormat instead")
  460. ACL_FUNC_VISIBILITY aclError aclSetTensorStorageFormat(aclTensorDesc *desc, aclFormat format);
  461. /**
  462. * @ingroup AscendCL
  463. * @brief Set the storage shape specified by the tensor description
  464. *
  465. * @param desc [OUT] pointer to the instance of aclTensorDesc
  466. * @param numDims [IN] the number of dimensions of the shape
  467. * @param dims [IN] the size of the specified dimension
  468. *
  469. * @retval ACL_SUCCESS The function is successfully executed.
  470. * @retval OtherValues Failure
  471. */
  472. ACL_DEPRECATED_MESSAGE("aclSetTensorStorageShape is deprecated, use aclSetTensorShape instead")
  473. ACL_FUNC_VISIBILITY aclError aclSetTensorStorageShape(aclTensorDesc *desc, int numDims, const int64_t *dims);
  474. /**
  475. * @ingroup AscendCL
  476. * @brief Set the format specified by the tensor description
  477. *
  478. * @param desc [OUT] pointer to the instance of aclTensorDesc
  479. * @param format [IN] the storage format
  480. *
  481. * @retval ACL_SUCCESS The function is successfully executed.
  482. * @retval OtherValues Failure
  483. */
  484. ACL_FUNC_VISIBILITY aclError aclSetTensorFormat(aclTensorDesc *desc, aclFormat format);
  485. /**
  486. * @ingroup AscendCL
  487. * @brief Set the shape specified by the tensor description
  488. *
  489. * @param desc [OUT] pointer to the instance of aclTensorDesc
  490. * @param numDims [IN] the number of dimensions of the shape
  491. * @param dims [IN] the size of the specified dimension
  492. *
  493. * @retval ACL_SUCCESS The function is successfully executed.
  494. * @retval OtherValues Failure
  495. */
  496. ACL_FUNC_VISIBILITY aclError aclSetTensorShape(aclTensorDesc *desc, int numDims, const int64_t *dims);
  497. /**
  498. * @ingroup AscendCL
  499. * @brief Set the original format specified by the tensor description
  500. *
  501. * @param desc [OUT] pointer to the instance of aclTensorDesc
  502. * @param format [IN] the storage format
  503. *
  504. * @retval ACL_SUCCESS The function is successfully executed.
  505. * @retval OtherValues Failure
  506. */
  507. ACL_FUNC_VISIBILITY aclError aclSetTensorOriginFormat(aclTensorDesc *desc, aclFormat format);
  508. /**
  509. * @ingroup AscendCL
  510. * @brief Set the original shape specified by the tensor description
  511. *
  512. * @param desc [OUT] pointer to the instance of aclTensorDesc
  513. * @param numDims [IN] the number of dimensions of the shape
  514. * @param dims [IN] the size of the specified dimension
  515. *
  516. * @retval ACL_SUCCESS The function is successfully executed.
  517. * @retval OtherValues Failure
  518. */
  519. ACL_FUNC_VISIBILITY aclError aclSetTensorOriginShape(aclTensorDesc *desc, int numDims, const int64_t *dims);
  520. /**
  521. * @ingroup AscendCL
  522. * @brief get op description info
  523. *
  524. * @param desc [IN] pointer to tensor description
  525. * @param index [IN] index of tensor
  526. *
  527. * @retval null for failed.
  528. * @retval OtherValues success.
  529. */
  530. ACL_FUNC_VISIBILITY aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index);
  531. /**
  532. * @ingroup AscendCL
  533. * @brief get address of tensor
  534. *
  535. * @param desc [IN] pointer to tensor description
  536. *
  537. * @retval null for failed
  538. * @retval OtherValues success
  539. */
  540. ACL_FUNC_VISIBILITY void *aclGetTensorDescAddress(const aclTensorDesc *desc);
  541. /**
  542. * @ingroup AscendCL
  543. * @brief Set the dynamic input name specified by the tensor description
  544. *
  545. * @param desc [OUT] pointer to the instance of aclTensorDesc
  546. * @param dynamicInputName [IN] pointer to the dynamic input name
  547. *
  548. * @retval ACL_SUCCESS The function is successfully executed.
  549. * @retval OtherValues Failure
  550. */
  551. ACL_FUNC_VISIBILITY aclError aclSetTensorDynamicInput(aclTensorDesc *desc, const char *dynamicInputName);
  552. /**
  553. * @ingroup AscendCL
  554. * @brief Set const data specified by the tensor description
  555. *
  556. * @param desc [OUT] pointer to the instance of aclTensorDesc
  557. * @param dataBuffer [IN] pointer to the const databuffer
  558. * @param length [IN] the length of const databuffer
  559. *
  560. * @retval ACL_SUCCESS The function is successfully executed.
  561. * @retval OtherValues Failure
  562. */
  563. ACL_FUNC_VISIBILITY aclError aclSetTensorConst(aclTensorDesc *desc, void *dataBuffer, size_t length);
  564. /**
  565. * @ingroup AscendCL
  566. * @brief Set tensor memory type specified by the tensor description
  567. *
  568. * @param desc [OUT] pointer to the instance of aclTensorDesc
  569. * @param memType [IN] ACL_MEMTYPE_DEVICE means device, ACL_MEMTYPE_HOST or
  570. * ACL_MEMTYPE_HOST_COMPILE_INDEPENDENT means host
  571. *
  572. * @retval ACL_SUCCESS The function is successfully executed.
  573. * @retval OtherValues Failure
  574. */
  575. ACL_FUNC_VISIBILITY aclError aclSetTensorPlaceMent(aclTensorDesc *desc, aclMemType memType);
  576. /**
  577. * @ingroup AscendCL
  578. * @brief an interface for users to output APP logs
  579. *
  580. * @param logLevel [IN] the level of current log
  581. * @param func [IN] the function where the log is located
  582. * @param file [IN] the file where the log is located
  583. * @param line [IN] Number of source lines where the log is located
  584. * @param fmt [IN] the format of current log
  585. * @param ... [IN] the value of current log
  586. */
  587. ACL_FUNC_VISIBILITY void aclAppLog(aclLogLevel logLevel, const char *func, const char *file, uint32_t line,
  588. const char *fmt, ...);
  589. /**
  590. * @ingroup AscendCL
  591. * @brief get soc name
  592. *
  593. * @retval null for failed
  594. * @retval OtherValues success
  595. */
  596. ACL_FUNC_VISIBILITY const char *aclrtGetSocName();
  597. #define ACL_APP_LOG(level, fmt, ...) aclAppLog(level, __FUNCTION__, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
  598. #ifdef __cplusplus
  599. }
  600. #endif
  601. #endif // INC_EXTERNAL_ACL_ACL_BASE_H_

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