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.

kernel.h 29 kB

5 years ago
5 years ago
3 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
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
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
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
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
3 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830
  1. /*
  2. * Copyright (c) Huawei Technologies Co., Ltd. 2020-2021. All rights reserved.
  3. * Description: kernel.h
  4. * Create: 2020-01-01
  5. */
  6. #ifndef CCE_RUNTIME_KERNEL_H
  7. #define CCE_RUNTIME_KERNEL_H
  8. #include "base.h"
  9. #include "stream.h"
  10. #if defined(__cplusplus)
  11. extern "C" {
  12. #endif
  13. /**
  14. * @ingroup rt_kernel
  15. * @brief shared memory data control
  16. */
  17. typedef struct tagRtSmData {
  18. uint64_t L2_mirror_addr; // preload or swap source address
  19. uint32_t L2_data_section_size; // every data size
  20. uint8_t L2_preload; // 1 - preload from mirrorAddr, 0 - no preload
  21. uint8_t modified; // 1 - data will be modified by kernel, 0 - no modified
  22. uint8_t priority; // data priority
  23. int8_t prev_L2_page_offset_base; // remap source section offset
  24. uint8_t L2_page_offset_base; // remap destination section offset
  25. uint8_t L2_load_to_ddr; // 1 - need load out, 0 - no need
  26. uint8_t reserved[2]; // reserved
  27. } rtSmData_t;
  28. /**
  29. * @ingroup rt_kernel
  30. * @brief shared memory description
  31. */
  32. typedef struct tagRtSmCtrl {
  33. rtSmData_t data[8]; // data description
  34. uint64_t size; // max page Num
  35. uint8_t remap[64]; /* just using for static remap mode, default:0xFF
  36. array index: virtual l2 page id, array value: physic l2 page id */
  37. uint8_t l2_in_main; // 0-DDR, 1-L2, default:0xFF
  38. uint8_t reserved[3];
  39. } rtSmDesc_t;
  40. typedef rtSmDesc_t rtL2Ctrl_t;
  41. /**
  42. * @ingroup rt_kernel
  43. * @brief device binary type
  44. */
  45. typedef struct tagRtDevBinary {
  46. uint32_t magic; // magic number
  47. uint32_t version; // version of binary
  48. const void *data; // binary data
  49. uint64_t length; // binary length
  50. } rtDevBinary_t;
  51. /**
  52. * @ingroup rt_kernel
  53. * @brief function mode type
  54. */
  55. #define ONLINE_PROF_MAX_PMU_NUM (8)
  56. typedef struct ProfilefDataInfo {
  57. const void *stubFunc;
  58. uint32_t blockDim;
  59. const void *args;
  60. uint32_t argsSize;
  61. rtSmDesc_t *smDesc;
  62. rtStream_t stream;
  63. uint64_t totalcycle;
  64. uint64_t ovcycle;
  65. uint64_t pmu_cnt[ONLINE_PROF_MAX_PMU_NUM];
  66. } rtProfDataInfo_t;
  67. /**
  68. * @ingroup rt_kernel
  69. * @brief function mode type
  70. */
  71. typedef enum {
  72. FUNC_MODE_NORMAL = 0,
  73. FUNC_MODE_PCTRACE_USERPROFILE_RECORDLOOP,
  74. FUNC_MODE_PCTRACE_USERPROFILE_SKIPLOOP,
  75. FUNC_MODE_PCTRACE_CYCLECNT_RECORDLOOP,
  76. FUNC_MODE_PCTRACE_CYCLECNT_SKIPLOOP,
  77. FUNC_MODE_BUTT
  78. } rtFuncModeType_t;
  79. /**
  80. * @ingroup rt_kernel
  81. * @brief kernel info
  82. */
  83. typedef struct rtKernelInfo {
  84. uint64_t task_offset; // kernel offset in module
  85. /* flowtable */
  86. void *arg; // launch kernel arg
  87. uint32_t arg_size;
  88. /* module */
  89. void *module_addr; // module::baseaddr_
  90. uint32_t module_size;
  91. } *rtKernelInfo_t;
  92. /**
  93. * @ingroup rt_kernel
  94. * @brief op name
  95. */
  96. typedef struct rtKernelLaunchNames {
  97. const char_t *soName; // defined for so name
  98. const char_t *kernelName; // defined for kernel type name
  99. const char_t *opName; // defined for operator name
  100. } rtKernelLaunchNames_t;
  101. /**
  102. * @ingroup rt_kernel
  103. * @brief args struct
  104. */
  105. typedef struct tagRtArgsWithTiling {
  106. void *args; // args host mem addr
  107. uint32_t argsSize; // input + output + tiling addr size + tiling data size
  108. uint32_t argsSizeWithoutTiling; // input + output + tiling addr size
  109. uint16_t tilingAddrOffset; // tiling addr offset
  110. uint16_t tilingDataOffset; // tiling data offset
  111. uint16_t hostInputAddrOffset; // index of host_memory input in inputs_addrs list
  112. uint16_t hostInputDataOffset; // host_mem input data offset
  113. uint8_t hasHostMemInput; // has host_memory input data in args or not: 0 means no host_memory input data,
  114. // others means has host_memory input data.
  115. uint8_t isNoNeedH2DCopy; // is no need host to device copy: 0 means need H2D copy,
  116. // others means doesn't need H2D copy.
  117. uint8_t reserved[6];
  118. } rtArgsWithTiling_t;
  119. /**
  120. * @ingroup rt_kernel
  121. * @brief host memory input struct
  122. */
  123. typedef struct rtHostInputInfo {
  124. uint16_t addrOffset;
  125. uint16_t dataOffset;
  126. } rtHostInputInfo_t;
  127. /**
  128. * @ingroup rt_kernel
  129. * @brief args struct
  130. */
  131. typedef struct tagRtArgsEx {
  132. void *args; // args host mem addr
  133. rtHostInputInfo_t *hostInputInfoPtr; // nullptr means no host mem input
  134. uint32_t argsSize; // input + output + tiling addr size + tiling data size + host mem
  135. uint16_t tilingAddrOffset; // tiling addr offset
  136. uint16_t tilingDataOffset; // tiling data offset
  137. uint16_t hostInputInfoNum; // hostInputInfo num
  138. uint8_t hasTiling; // if has tiling: 0 means no tiling
  139. uint8_t isNoNeedH2DCopy; // is no need host to device copy: 0 means need H2D copy,
  140. // others means doesn't need H2D copy.
  141. uint8_t reserved[4];
  142. } rtArgsEx_t;
  143. /**
  144. * @ingroup rt_KernelConfigDump
  145. * @brief device dump type
  146. */
  147. typedef enum tagRtDumpKind {
  148. RT_DATA_DUMP_KIND_INVALID = -1,
  149. RT_DATA_DUMP_KIND_DUMP = 0,
  150. RT_DATA_DUMP_KIND_RESERVED = 1,
  151. } rtDumpKind_t;
  152. /**
  153. * @ingroup rt_kernel
  154. * @brief report callback
  155. */
  156. typedef rtError_t (*rtKernelReportCallback)(rtStream_t stm, rtKernelInfo_t kernelInfo);
  157. /**
  158. * @ingroup rt_kernel
  159. * @brief stream report callback
  160. */
  161. typedef void (*rtCallback_t)(void *fnData);
  162. /**
  163. * @ingroup rt_kernel
  164. * @brief magic number of plain binary for aicore
  165. */
  166. #define RT_DEV_BINARY_MAGIC_PLAIN 0xabceed50U
  167. /**
  168. * @ingroup rt_kernel
  169. * @brief magic number of plain binary for aicpu
  170. */
  171. #define RT_DEV_BINARY_MAGIC_PLAIN_AICPU 0xabceed51U
  172. /**
  173. * @ingroup rt_kernel
  174. * @brief magic number of plain binary for aivector
  175. */
  176. #define RT_DEV_BINARY_MAGIC_PLAIN_AIVEC 0xabceed52U
  177. /**
  178. * @ingroup rt_kernel
  179. * @brief magic number of elf binary for aicore
  180. */
  181. #define RT_DEV_BINARY_MAGIC_ELF 0x43554245U
  182. /**
  183. * @ingroup rt_kernel
  184. * @brief magic number of elf binary for aicpu
  185. */
  186. #define RT_DEV_BINARY_MAGIC_ELF_AICPU 0x41415243U
  187. /**
  188. * @ingroup rt_kernel
  189. * @brief magic number of elf binary for aivector
  190. */
  191. #define RT_DEV_BINARY_MAGIC_ELF_AIVEC 0x41415246U
  192. /**
  193. * @ingroup rt_kernel
  194. * @brief magic number of elf binary for aicube
  195. */
  196. #define RT_DEV_BINARY_MAGIC_ELF_AICUBE 0x41494343U
  197. /**
  198. * @ingroup rt_kernel_flags
  199. * @brief kernel op bit flags
  200. */
  201. #define RT_KERNEL_DEFAULT (0x00U)
  202. #define RT_KERNEL_CONVERT (0x01U)
  203. #define RT_KERNEL_DUMPFLAG (0x02U)
  204. #define RT_FUSION_KERNEL_DUMPFLAG (0x04U)
  205. #define RT_KERNEL_CUSTOM_AICPU (0x08U)
  206. // STARS topic scheduler sqe : topic_type
  207. #define RT_KERNEL_DEVICE_FIRST (0x10U)
  208. #define RT_KERNEL_HOST_ONLY (0x20U)
  209. #define RT_KERNEL_HOST_FIRST (0x40U)
  210. /**
  211. * @ingroup rt_kernel
  212. * @brief kernel mode
  213. **/
  214. #define RT_DEFAULT_KERNEL_MODE (0x00U)
  215. #define RT_NORMAL_KERNEL_MODE (0x01U)
  216. #define RT_ALL_KERNEL_MODE (0x02U)
  217. /**
  218. * @ingroup rt_kernel
  219. * @brief SHAPE kernel type
  220. **/
  221. #define RT_STATIC_SHAPE_KERNEL (0x00U)
  222. #define RT_DYNAMIC_SHAPE_KERNEL (0x01U)
  223. /**
  224. * @ingroup rt_kernel
  225. * @brief kernel L1 Fusion Dump bit flags
  226. */
  227. #define RT_DDR_ADDR (0x0U)
  228. /**
  229. * @ingroup rt_kernel
  230. * @brief register device binary
  231. * @param [in] bin device binary description
  232. * @param [out] hdl device binary handle
  233. * @return RT_ERROR_NONE for ok
  234. * @return RT_ERROR_INVALID_VALUE for error input
  235. */
  236. RTS_API rtError_t rtDevBinaryRegister(const rtDevBinary_t *bin, void **hdl);
  237. /**
  238. * @ingroup rt_kernel
  239. * @brief register device binary with all kernel
  240. * @param [in] bin device binary description
  241. * @param [out] hdl device binary handle
  242. * @return RT_ERROR_NONE for ok
  243. * @return RT_ERROR_INVALID_VALUE for error input
  244. */
  245. RTS_API rtError_t rtRegisterAllKernel(const rtDevBinary_t *bin, void **hdl);
  246. /**
  247. * @ingroup rt_kernel
  248. * @brief register fast memeory device binary
  249. * @param [in] hdl device binary handle
  250. * @return RT_ERROR_NONE for ok
  251. * @return RT_ERROR_INVALID_VALUE for error input
  252. */
  253. RTS_API rtError_t rtBinaryRegisterToFastMemory(void *hdl);
  254. /**
  255. * @ingroup rt_kernel
  256. * @brief unregister device binary
  257. * @param [in] hdl device binary handle
  258. * @return RT_ERROR_NONE for ok
  259. * @return RT_ERROR_INVALID_VALUE for error input
  260. */
  261. RTS_API rtError_t rtDevBinaryUnRegister(void *hdl);
  262. /**
  263. * @ingroup rt_kernel
  264. * @brief register device binary metadata
  265. * @param [in] hdl device binary description
  266. * @param [in] metadata device binary metadata
  267. * @return RT_ERROR_NONE for ok
  268. * @return RT_ERROR_INVALID_VALUE for error input
  269. */
  270. RTS_API rtError_t rtMetadataRegister(void *hdl, const char_t *metadata);
  271. /**
  272. * @ingroup rt_kernel
  273. * @brief register device binary dependency
  274. * @param [in] mHandle master device binary description
  275. * @param [in] sHandle slave device binary description
  276. * @return RT_ERROR_NONE for ok
  277. * @return RT_ERROR_INVALID_VALUE for error input
  278. */
  279. RTS_API rtError_t rtDependencyRegister(void *mHandle, void *sHandle);
  280. /**
  281. * @ingroup rt_kernel
  282. * @brief register device function
  283. * @param [in] binHandle device binary handle
  284. * @param [in] stubFunc stub function
  285. * @param [in] stubName stub function name
  286. * @param [in] kernelInfoExt kernel Info extension. device function description or tiling key,
  287. * depending static shape or dynmaic shape.
  288. * @return RT_ERROR_NONE for ok
  289. * @return RT_ERROR_INVALID_VALUE for error input
  290. */
  291. RTS_API rtError_t rtFunctionRegister(void *binHandle, const void *stubFunc, const char_t *stubName,
  292. const void *kernelInfoExt, uint32_t funcMode);
  293. /**
  294. * @ingroup rt_kernel
  295. * @brief find stub function by name
  296. * @param [in] stubName stub function name
  297. * @param [out] stubFunc stub function
  298. * @return RT_ERROR_NONE for ok
  299. * @return RT_ERROR_INVALID_VALUE for error input
  300. */
  301. RTS_API rtError_t rtGetFunctionByName(const char_t *stubName, void **stubFunc);
  302. /**
  303. * @ingroup rt_kernel
  304. * @brief find addr by stub func
  305. * @param [in] stubFunc stub function
  306. * @param [out] addr
  307. * @return RT_ERROR_NONE for ok
  308. * @return RT_ERROR_INVALID_VALUE for error input
  309. */
  310. RTS_API rtError_t rtGetAddrByFun(const void *stubFunc, void **addr);
  311. /**
  312. * @ingroup rt_kernel
  313. * @brief query registered or not by stubName
  314. * @param [in] stubName stub function name
  315. * @return RT_ERROR_NONE for ok
  316. * @return RT_ERROR_INVALID_VALUE for error input
  317. */
  318. RTS_API rtError_t rtQueryFunctionRegistered(const char_t *stubName);
  319. /**
  320. * @ingroup rt_kernel
  321. * @brief config data dump
  322. * @param [in] dumpSizePerBlock dump size
  323. * @param [in] blockDim block dimentions
  324. * @param [in] dumpBaseAddr dump base address
  325. * @return RT_ERROR_NONE for ok
  326. * @return RT_ERROR_INVALID_VALUE for error input
  327. */
  328. RTS_API rtError_t rtKernelConfigDump(uint32_t kind, uint32_t dumpSizePerBlock, uint32_t blockDim, void **dumpBaseAddr,
  329. rtStream_t stm);
  330. /**
  331. * @ingroup rt_kernel
  332. * @brief get kernel address and prefetchCnt
  333. * @param [in] hdl program for dynamic shape
  334. * @param [in] tilingKey tilingKey for dynamic shape
  335. * @param [in] stubFunc stubFunc for static shape
  336. * @param [in] flag flag for distinguishing between dynamic shape and static shape
  337. * @param [out] addr address of kernel function
  338. * @param [out] prefetchCnt prefetchCnt of kernel function
  339. * @return RT_ERROR_NONE for ok
  340. * @return RT_ERROR_INVALID_VALUE for error input
  341. */
  342. RTS_API rtError_t rtKernelGetAddrAndPrefCnt(void *hdl, const uint64_t tilingKey, const void * const stubFunc,
  343. const uint32_t flag, void **addr, uint32_t *prefetchCnt);
  344. /**
  345. * @ingroup rt_kernel
  346. * @brief launch kernel to device
  347. * @param [in] stubFunc stub function
  348. * @param [in] blockDim block dimentions
  349. * @param [in] args argments address for kernel function
  350. * @param [in] argsSize argements size
  351. * @param [in] smDesc shared memory description
  352. * @param [in] stm associated stream
  353. * @return RT_ERROR_NONE for ok
  354. * @return RT_ERROR_INVALID_VALUE for error input
  355. */
  356. RTS_API rtError_t rtKernelLaunch(const void *stubFunc, uint32_t blockDim, void *args, uint32_t argsSize,
  357. rtSmDesc_t *smDesc, rtStream_t stm);
  358. /**
  359. * @ingroup rt_kernel
  360. * @brief launch kernel with handle to device
  361. * @param [in] hdl program
  362. * @param [in] kernelInfoExt kernel Info extension. device function description or tiling key,
  363. * depending static shape or dynmaic shape.
  364. * @param [in] blockDim block dimentions
  365. * @param [in] argsInfo argments address for kernel function
  366. * @param [in] smDesc shared memory description
  367. * @param [in] stm associated stream
  368. * @param [in] kernelInfo kernel info
  369. * @return RT_ERROR_NONE for ok
  370. * @return RT_ERROR_INVALID_VALUE for error input
  371. */
  372. RTS_API rtError_t rtKernelLaunchWithHandle(void *hdl, const void *kernelInfoExt, uint32_t blockDim,
  373. rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  374. const void *kernelInfo);
  375. /**
  376. * @ingroup rtKernelLaunchWithHandleV2
  377. * @brief launch kernel with handle to device
  378. * @param [in] hdl program
  379. * @param [in] kernelInfoExt kernel Info extension. device function description or tiling key,
  380. * depending static shape or dynmaic shape.
  381. * @param [in] blockDim block dimentions
  382. * @param [in] argsInfo argments address for kernel function
  383. * @param [in] smDesc shared memory description
  384. * @param [in] stm associated stream
  385. * @param [in] kernelInfo kernel info
  386. * @return RT_ERROR_NONE for ok
  387. * @return RT_ERROR_INVALID_VALUE for error input
  388. */
  389. RTS_API rtError_t rtKernelLaunchWithHandleV2(void *hdl, const void *kernelInfoExt, uint32_t blockDim,
  390. rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  391. const void *kernelInfo);
  392. /**
  393. * @ingroup rtKernelLaunchWithFlag
  394. * @brief launch kernel to device
  395. * @param [in] stubFunc stub function
  396. * @param [in] blockDim block dimentions
  397. * @param [in] argsInfo argments address for kernel function
  398. * @param [in] smDesc shared memory description
  399. * @param [in] stm associated stream
  400. * @param [in] flags dump flag
  401. * @return RT_ERROR_NONE for ok
  402. * @return RT_ERROR_INVALID_VALUE for error input
  403. */
  404. RTS_API rtError_t rtKernelLaunchWithFlag(const void *stubFunc, uint32_t blockDim, rtArgsEx_t *argsInfo,
  405. rtSmDesc_t *smDesc, rtStream_t stm, uint32_t flags);
  406. /**
  407. * @ingroup rtKernelLaunchWithFlagV2
  408. * @brief launch kernel to device
  409. * @param [in] stubFunc stub function
  410. * @param [in] blockDim block dimentions
  411. * @param [in] argsInfo argments address for kernel function
  412. * @param [in] smDesc shared memory description
  413. * @param [in] stm associated stream
  414. * @param [in] flags dump flag
  415. * @return RT_ERROR_NONE for ok
  416. * @return RT_ERROR_INVALID_VALUE for error input
  417. */
  418. RTS_API rtError_t rtKernelLaunchWithFlagV2(const void *stubFunc, uint32_t blockDim, rtArgsEx_t *argsInfo,
  419. rtSmDesc_t *smDesc, rtStream_t stm, uint32_t flags);
  420. /**
  421. * @ingroup rt_kernel(abandoned)
  422. * @brief launch kernel to device
  423. * @param [in] args argments address for kernel function
  424. * @param [in] argsSize argements size
  425. * @param [in] flags launch flags
  426. * @param [in] stm associated stream
  427. * @return RT_ERROR_NONE for ok
  428. * @return RT_ERROR_INVALID_VALUE for error input
  429. */
  430. RTS_API rtError_t rtKernelLaunchEx(void *args, uint32_t argsSize, uint32_t flags, rtStream_t stm);
  431. /**
  432. * @ingroup rt_kernel(in use)
  433. * @brief launch kernel to device
  434. * @param [in] opName opkernel name
  435. * @param [in] args argments address for kernel function
  436. * @param [in] argsSize argements size
  437. * @param [in] flags launch flags
  438. * @param [in] stm associated stream
  439. * @return RT_ERROR_NONE for ok
  440. * @return RT_ERROR_INVALID_VALUE for error input
  441. */
  442. RTS_API rtError_t rtKernelLaunchFwk(const char_t *opName, void *args, uint32_t argsSize, uint32_t flags,
  443. rtStream_t rtStream);
  444. /**
  445. * @ingroup rt_kernel(abandoned)
  446. * @brief launch cpu kernel to device
  447. * @param [in] soName so name
  448. * @param [in] kernelName kernel name
  449. * @param [in] blockDim block dimentions
  450. * @param [in] args argments address for kernel function
  451. * @param [in] argsSize argments size
  452. * @param [in] smDesc shared memory description
  453. * @param [in] stm associated stream
  454. * @return RT_ERROR_NONE for ok
  455. * @return RT_ERROR_INVALID_VALUE for error input
  456. */
  457. RTS_API rtError_t rtCpuKernelLaunch(const void *soName, const void *kernelName, uint32_t blockDim, const void *args,
  458. uint32_t argsSize, rtSmDesc_t *smDesc, rtStream_t stm);
  459. /**
  460. * @ingroup rt_kernel(in use)
  461. * @brief launch cpu kernel to device
  462. * @param [in] launchNames names for kernel launch
  463. * @param [in] blockDim block dimentions
  464. * @param [in] args argments address for kernel function
  465. * @param [in] argsSize argments size
  466. * @param [in] smDesc shared memory description
  467. * @param [in] stm associated stream
  468. * @return RT_ERROR_NONE for ok
  469. * @return RT_ERROR_INVALID_VALUE for error input
  470. */
  471. RTS_API rtError_t rtAicpuKernelLaunch(const rtKernelLaunchNames_t *launchNames,
  472. uint32_t blockDim, const void *args, uint32_t argsSize, rtSmDesc_t *smDesc, rtStream_t stm);
  473. /**
  474. * @ingroup rtCpuKernelLaunchWithFlag(abandoned)
  475. * @brief launch cpu kernel to device with dump identifier
  476. * @param [in] soName so name
  477. * @param [in] kernelName kernel name
  478. * @param [in] blockDim block dimentions
  479. * @param [in] argsInfo argments address for kernel function
  480. * @param [in] smDesc shared memory description
  481. * @param [in] stm associated stream
  482. * @param [in] flag dump flag or others function flag
  483. * @return RT_ERROR_NONE for ok
  484. * @return RT_ERROR_INVALID_VALUE for error input
  485. */
  486. RTS_API rtError_t rtCpuKernelLaunchWithFlag(const void *soName, const void *kernelName, uint32_t blockDim,
  487. const rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  488. uint32_t flags);
  489. /**
  490. * @ingroup rtCpuKernelLaunchWithFlagV2
  491. * @brief launch cpu kernel to device with dump identifier
  492. * @param [in] soName so name
  493. * @param [in] kernelName kernel name
  494. * @param [in] blockDim block dimentions
  495. * @param [in] argsInfo argments address for kernel function
  496. * @param [in] smDesc shared memory description
  497. * @param [in] stm associated stream
  498. * @param [in] flags dump flag or others function flag
  499. * @return RT_ERROR_NONE for ok
  500. * @return RT_ERROR_INVALID_VALUE for error input
  501. */
  502. RTS_API rtError_t rtCpuKernelLaunchWithFlagV2(const void *soName, const void *kernelName, uint32_t blockDim,
  503. const rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  504. uint32_t flags);
  505. /**
  506. * @ingroup rtAicpuKernelLaunchWithFlag(in use)
  507. * @brief launch cpu kernel to device with dump identifier
  508. * @param [in] launchNames names for kernel launch
  509. * @param [in] blockDim block dimentions
  510. * @param [in] args argments address for kernel function
  511. * @param [in] smDesc shared memory description
  512. * @param [in] stm associated stream
  513. * @param [in] flags dump flag or others function flag
  514. * @return RT_ERROR_NONE for ok
  515. * @return RT_ERROR_INVALID_VALUE for error input
  516. */
  517. RTS_API rtError_t rtAicpuKernelLaunchWithFlag(const rtKernelLaunchNames_t *launchNames, uint32_t blockDim,
  518. const rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  519. uint32_t flags);
  520. /**
  521. * @ingroup rtAicpuKernelLaunchWithFlagV2(in use)
  522. * @brief launch cpu kernel to device with dump identifier
  523. * @param [in] launchNames names for kernel launch
  524. * @param [in] blockDim block dimentions
  525. * @param [in] argsInfo argments address for kernel function
  526. * @param [in] smDesc shared memory description
  527. * @param [in] stm associated stream
  528. * @param [in] flags dump flag or others function flag
  529. * @return RT_ERROR_NONE for ok
  530. * @return RT_ERROR_INVALID_VALUE for error input
  531. */
  532. RTS_API rtError_t rtAicpuKernelLaunchWithFlagV2(const rtKernelLaunchNames_t *launchNames, uint32_t blockDim,
  533. const rtArgsEx_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm,
  534. uint32_t flags);
  535. /**
  536. * @ingroup rt_kernel
  537. * @brief L1 fusion dump addr transfered to device
  538. * @param [in] mdl handle info
  539. * @param [in] addr ddr address of L1 Fusion Dump
  540. * @param [in] dumpSize memory size
  541. * @param [in] flag memory flag
  542. * @return RT_ERROR_NONE for ok
  543. * @return RT_ERROR_INVALID_VALUE for error input
  544. */
  545. RTS_API rtError_t rtDumpAddrSet(rtModel_t mdl, void *addr, uint32_t dumpSize, uint32_t flag);
  546. /**
  547. * @ingroup rt_kernel
  548. * @brief load dump info to aicpu
  549. * @param [in] dumpInfo dump info
  550. * @param [in] length length of dump info
  551. * @return RT_ERROR_NONE for ok
  552. * @return RT_ERROR_INVALID_VALUE for error input
  553. */
  554. RTS_API rtError_t rtDatadumpInfoLoad(const void *dumpInfo, uint32_t length);
  555. /**
  556. * @ingroup rt_kernel
  557. * @brief launch npu get float status task
  558. * @param [in] outputAddr pointer to op output addr
  559. * @param [in] outputSize op output size
  560. * @param [in] checkMode check mode
  561. * @param [in] stm associated stream
  562. * @return RT_ERROR_NONE for ok
  563. * @return RT_ERROR_INVALID_VALUE for error input
  564. */
  565. RTS_API rtError_t rtNpuGetFloatStatus(void *outputAddr, uint64_t outputSize, uint32_t checkMode, rtStream_t stm);
  566. /**
  567. * @ingroup rt_kernel
  568. * @brief launch npu clear float status task
  569. * @param [in] checkMode check mode
  570. * @param [in] stm associated stream
  571. * @return RT_ERROR_NONE for ok
  572. * @return RT_ERROR_INVALID_VALUE for error input
  573. */
  574. RTS_API rtError_t rtNpuClearFloatStatus(uint32_t checkMode, rtStream_t stm);
  575. #ifndef __CLANG_CCE_RUNTIME_H__
  576. #define __CLANG_CCE_RUNTIME_H__
  577. /**
  578. * @ingroup rt_kernel
  579. * @brief configure call argment for next rtLaunch in current thread
  580. * @param [in] numBlocks block dimentions
  581. * @param [in] smDesc shared memory description
  582. * @param [in] stm associated stream
  583. * @return RT_ERROR_NONE for ok
  584. * @return RT_ERROR_INVALID_VALUE for error input
  585. */
  586. #ifdef __cplusplus
  587. RTS_API rtError_t rtConfigureCall(uint32_t numBlocks, rtSmDesc_t *smDesc = nullptr, rtStream_t stm = nullptr);
  588. #else
  589. RTS_API rtError_t rtConfigureCall(uint32_t numBlocks, rtSmDesc_t *smDesc, rtStream_t stm);
  590. #endif
  591. #endif // __CLANG_CCE_RUNTIME_H__
  592. /**
  593. * @ingroup rt_kernel
  594. * @brief setup argment for next rtLaunch in current thread
  595. * @param [in] args argment address for kernel function
  596. * @param [in] size argment size
  597. * @param [in] offset argment table offset
  598. * @return RT_ERROR_NONE for ok
  599. * @return RT_ERROR_INVALID_VALUE for error input
  600. */
  601. RTS_API rtError_t rtSetupArgument(const void *args, uint32_t size, uint32_t offset);
  602. /**
  603. * @ingroup rt_kernel
  604. * @brief launch kernel to device with previous setting kernel argment
  605. * and call argment
  606. * @param [in] stubFunc stub function
  607. * @return RT_ERROR_NONE for ok
  608. * @return RT_ERROR_INVALID_VALUE for error input
  609. */
  610. RTS_API rtError_t rtLaunch(const void *stubFunc);
  611. /**
  612. * @ingroup rt_kernel
  613. * @brief implicitly transfered data to device.
  614. * lifecycle end after next kernel task finish
  615. * @param [in] ptr host memory
  616. * @param [in] size host memory size
  617. * @param [in] flag reserved. set to 0
  618. * @param [out] args returned arg. used for next kernel's arg.
  619. * @return RT_ERROR_NONE for ok
  620. * @return RT_ERROR_INVALID_VALUE for error input
  621. */
  622. RTS_API rtError_t rtKernelConfigTransArg(const void *ptr, uint64_t size, uint32_t flag, void **args);
  623. /**
  624. * @ingroup rt_kernel
  625. * @brief start fusion kernels.
  626. * @param [in] stm stream for fusion kernels
  627. * @return RT_ERROR_NONE for ok
  628. * @return RT_ERROR_INVALID_VALUE for error input
  629. */
  630. RTS_API rtError_t rtKernelFusionStart(rtStream_t stm);
  631. /**
  632. * @ingroup rt_kernel
  633. * @brief end fusion kernels.
  634. * @param [in] stm stream for fusion kernels
  635. * @return RT_ERROR_NONE for ok
  636. * @return RT_ERROR_INVALID_VALUE for error input
  637. */
  638. RTS_API rtError_t rtKernelFusionEnd(rtStream_t stm);
  639. /**
  640. * @ingroup rt_kernel
  641. * @brief set kernelinfo callback
  642. * @param [in] callback
  643. * @return RT_ERROR_NONE for ok
  644. * @return RT_ERROR_INVALID_VALUE for error input
  645. */
  646. RTS_API rtError_t rtSetKernelReportCallback(rtKernelReportCallback callBack);
  647. /**
  648. * @ingroup rt_kernel
  649. * @brief subscribe stream callback report.
  650. * @param [in] threadId thread id for stream
  651. * @param [in] stm stream for subscribe
  652. * @return RT_ERROR_NONE for ok
  653. * @return RT_ERROR_INVALID_VALUE for error input
  654. */
  655. RTS_API rtError_t rtSubscribeReport(uint64_t threadId, rtStream_t stm);
  656. /**
  657. * @ingroup rt_kernel
  658. * @brief add callback launch task in stream.
  659. * @param [in] callBackFunc app callback function
  660. * @param [in] fnData user data
  661. * @param [in] stm subscribed stream
  662. * @return RT_ERROR_NONE for ok
  663. * @return RT_ERROR_INVALID_VALUE for error input
  664. */
  665. RTS_API rtError_t rtCallbackLaunch(rtCallback_t callBackFunc, void *fnData, rtStream_t stm, bool isBlock);
  666. /**
  667. * @ingroup rt_kernel
  668. * @brief process callback report.
  669. * @param [in] timeout if timeout=-1, while(1); else timeout
  670. * @return RT_ERROR_NONE for ok
  671. * @return RT_ERROR_INVALID_VALUE for error input
  672. */
  673. RTS_API rtError_t rtProcessReport(int32_t timeout);
  674. /**
  675. * @ingroup rt_kernel
  676. * @brief unsubscribe callback report.
  677. * @param [in] threadId thread id for stream
  678. * @param [in] stm stream for subscribe
  679. * @return RT_ERROR_NONE for ok
  680. * @return RT_ERROR_INVALID_VALUE for error input
  681. */
  682. RTS_API rtError_t rtUnSubscribeReport(uint64_t threadId, rtStream_t stm);
  683. /**
  684. * @ingroup profiling_base
  685. * @brief start online prof.
  686. * @return RT_ERROR_NONE for ok
  687. * @return RT_ERROR_INVALID_VALUE for error input
  688. */
  689. RTS_API rtError_t rtStartOnlineProf(rtStream_t stm, uint32_t sampleNum);
  690. /**
  691. * @ingroup profiling_base
  692. * @brief stop online prof.
  693. * @return RT_ERROR_NONE for ok
  694. * @return RT_ERROR_INVALID_VALUE for error input
  695. */
  696. RTS_API rtError_t rtStopOnlineProf(rtStream_t stm);
  697. /**
  698. * @ingroup profiling_base
  699. * @brief get online prof.
  700. * @return RT_ERROR_NONE for ok
  701. * @return RT_ERROR_INVALID_VALUE for error input
  702. */
  703. RTS_API rtError_t rtGetOnlineProfData(rtStream_t stm, rtProfDataInfo_t *pProfData, uint32_t profDataNum);
  704. /**
  705. * @ingroup profiling_base
  706. * @brief start mdc profiler.
  707. * @return RT_ERROR_NONE for ok
  708. * @return RT_ERROR_INVALID_VALUE for error input
  709. */
  710. RTS_API rtError_t rtStartMDCProfiler(void **addr, uint32_t length);
  711. /**
  712. * @ingroup profiling_base
  713. * @brief stop mdc profiler.
  714. * @return RT_ERROR_NONE for ok
  715. * @return RT_ERROR_INVALID_VALUE for error input
  716. */
  717. RTS_API rtError_t rtStopMDCProfiler(void *addr);
  718. /**
  719. * @ingroup rt_kernel
  720. * @brief launch kernel with tiling data to device
  721. * @param [in] stubFunc stub function
  722. * @param [in] blockDim block dimentions
  723. * @param [in] argsInfo argments info address for kernel function
  724. * @param [in] smDesc shared memory description
  725. * @param [in] stm associated stream
  726. * @return RT_ERROR_NONE for ok
  727. * @return RT_ERROR_INVALID_VALUE for error input
  728. */
  729. RTS_API rtError_t rtKernelLaunchWithTiling(const void *stubFunc, uint32_t blockDim,
  730. rtArgsWithTiling_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm);
  731. /**
  732. * @ingroup rt_kernel
  733. * @brief launch kernel with handle and tiling data to device
  734. * @param [in] hdl program
  735. * @param [in] kernelInfoExt kernel Info extension. device function description or tiling key,
  736. * depending static shape or dynmaic shape.
  737. * @param [in] blockDim block dimentions
  738. * @param [in] argsInfo argments info address for kernel function
  739. * @param [in] smDesc shared memory description
  740. * @param [in] stm associated stream
  741. * @param [in] kernelInfo kernel info
  742. * @return RT_ERROR_NONE for ok
  743. * @return RT_ERROR_INVALID_VALUE for error input
  744. */
  745. RTS_API rtError_t rtKernelLaunchWithHandleAndTiling(void *hdl, const void *kernelInfoExt, uint32_t blockDim,
  746. rtArgsWithTiling_t *argsInfo, rtSmDesc_t *smDesc, rtStream_t stm, const void* kernelInfo);
  747. #if defined(__cplusplus)
  748. }
  749. #endif
  750. #endif // CCE_RUNTIME_KERNEL_H

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