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 23 kB

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

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