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.

runtime_stub.cc 14 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
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
4 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
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471
  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. #include <cce/dnn.h>
  17. #include <securec.h>
  18. #ifdef __cplusplus
  19. extern "C" {
  20. #endif
  21. #define EVENT_LENTH 10
  22. rtError_t rtCtxSetCurrent(rtContext_t ctx) { return RT_ERROR_NONE; }
  23. rtError_t rtGetStreamId(rtStream_t stream, int32_t *stream_id) {
  24. *stream_id = 0;
  25. return RT_ERROR_NONE;
  26. }
  27. rtError_t rtCtxGetCurrent(rtContext_t *ctx) {
  28. uintptr_t x = 1;
  29. *ctx = (rtContext_t *)x;
  30. return RT_ERROR_NONE;
  31. }
  32. rtError_t rtCtxSetDryRun(rtContext_t ctx, rtDryRunFlag_t enable, uint32_t flag) { return RT_ERROR_NONE; }
  33. rtError_t rtEventGetTimeStamp(uint64_t *time, rtEvent_t event) {
  34. *time = 12345;
  35. return RT_ERROR_NONE;
  36. }
  37. rtError_t rtEventCreate(rtEvent_t *event) {
  38. *event = new int[EVENT_LENTH];
  39. return RT_ERROR_NONE;
  40. }
  41. rtError_t rtEventCreateWithFlag(rtEvent_t *event, uint32_t flag) {
  42. return rtEventCreate(event);
  43. }
  44. rtError_t rtEventRecord(rtEvent_t event, rtStream_t stream) { return RT_ERROR_NONE; }
  45. rtError_t rtEventSynchronize(rtEvent_t event) { return RT_ERROR_NONE; }
  46. rtError_t rtEventDestroy(rtEvent_t event) {
  47. delete[](int *) event;
  48. return RT_ERROR_NONE;
  49. }
  50. rtError_t rtMalloc(void **dev_ptr, uint64_t size, rtMemType_t type) {
  51. *dev_ptr = new uint8_t[size];
  52. return RT_ERROR_NONE;
  53. }
  54. rtError_t rtMemset(void *dev_ptr, uint64_t dest_max, uint32_t value, uint64_t count) { return RT_ERROR_NONE; }
  55. rtError_t rtFree(void *dev_ptr) {
  56. delete[](uint8_t *) dev_ptr;
  57. return RT_ERROR_NONE;
  58. }
  59. rtError_t rtMallocHost(void **host_ptr, uint64_t size) {
  60. *host_ptr = new uint8_t[size];
  61. return RT_ERROR_NONE;
  62. }
  63. rtError_t rtFreeHost(void *host_ptr) {
  64. delete[](uint8_t *) host_ptr;
  65. return RT_ERROR_NONE;
  66. }
  67. rtError_t rtStreamCreate(rtStream_t *stream, int32_t priority) {
  68. *stream = new uint32_t;
  69. return RT_ERROR_NONE;
  70. }
  71. rtError_t rtStreamDestroy(rtStream_t stream) {
  72. if (stream != nullptr) {
  73. delete (uint32_t *)stream;
  74. }
  75. return RT_ERROR_NONE;
  76. }
  77. rtError_t rtSetDevice(int32_t device) { return RT_ERROR_NONE; }
  78. rtError_t rtStreamSynchronize(rtStream_t stream) { return RT_ERROR_NONE; }
  79. rtError_t rtMemcpy(void *dst, uint64_t dest_max, const void *src, uint64_t count, rtMemcpyKind_t kind) {
  80. if (dst != nullptr && src != nullptr) {
  81. memcpy_s(dst, dest_max, src, count);
  82. }
  83. return RT_ERROR_NONE;
  84. }
  85. rtError_t rtMemcpyAsync(void *dst, uint64_t dest_max, const void *src, uint64_t count, rtMemcpyKind_t kind,
  86. rtStream_t stream) {
  87. if (dst != nullptr && src != nullptr) {
  88. memcpy_s(dst, dest_max, src, count);
  89. }
  90. return RT_ERROR_NONE;
  91. }
  92. rtError_t rtStreamWaitEvent(rtStream_t stream, rtEvent_t event) { return RT_ERROR_NONE; }
  93. rtError_t rtSetTSDevice(uint32_t tsId) {
  94. return RT_ERROR_NONE;
  95. }
  96. rtError_t rtGetDeviceCount(int32_t *count) {
  97. *count = 1;
  98. return RT_ERROR_NONE;
  99. }
  100. rtError_t rtDeviceReset(int32_t device) { return RT_ERROR_NONE; }
  101. rtError_t rtEventElapsedTime(float *time, rtEvent_t start, rtEvent_t end) {
  102. *time = 10.0f;
  103. return RT_ERROR_NONE;
  104. }
  105. rtError_t rtFunctionRegister(void *bin_handle, const void *stub_func, const char *stub_name, const void *dev_func,
  106. uint32_t func_mode) {
  107. return RT_ERROR_NONE;
  108. }
  109. rtError_t rtDevBinaryRegister(const rtDevBinary_t *bin, void **handle) { return RT_ERROR_NONE; }
  110. rtError_t rtRegisterAllKernel(const rtDevBinary_t *bin, void **handle) { return RT_ERROR_NONE; }
  111. rtError_t rtKernelConfigTransArg(const void *ptr, uint64_t size, uint32_t flag, void **arg) { return RT_ERROR_NONE; }
  112. rtError_t rtKernelLaunchWithHandle(void *handle, const void *devFunc, uint32_t blockDim, void *args, uint32_t argsSize,
  113. rtSmDesc_t *smDesc, rtStream_t stream, const void *kernelInfo) {
  114. return RT_ERROR_NONE;
  115. }
  116. rtError_t rtKernelLaunch(const void *stub_func, uint32_t block_dim, void *args, uint32_t args_size, rtSmDesc_t *sm_desc,
  117. rtStream_t stream) {
  118. return RT_ERROR_NONE;
  119. }
  120. rtError_t rtSetupArgument(const void *arg, uint32_t size, uint32_t offset) { return RT_ERROR_NONE; }
  121. rtError_t rtLaunch(const void *stub_func) { return RT_ERROR_NONE; }
  122. rtError_t rtDevBinaryUnRegister(void *handle) { return RT_ERROR_NONE; }
  123. rtError_t rtConfigureCall(uint32_t num_blocks, rtSmDesc_t *sm_desc, rtStream_t stream) { return RT_ERROR_NONE; }
  124. rtError_t rtSetProfDir(char *prof_dir) { return RT_ERROR_NONE; }
  125. rtError_t rtSetProfDirEx(const char *profDir, const char *address, const char *jobCtx) { return RT_ERROR_NONE; }
  126. rtError_t rtAiCoreMemorySizes(rtAiCoreMemorySize_t *aicore_memory_size) { return RT_ERROR_NONE; }
  127. rtError_t rtSetKernelReportCallback(rtKernelReportCallback callback) {
  128. rtKernelInfo rt_kernel_info = {0};
  129. rt_kernel_info.arg_size = 12;
  130. rt_kernel_info.task_offset = 100;
  131. rt_kernel_info.arg = (void *)100;
  132. rt_kernel_info.module_addr = (void *)100;
  133. rt_kernel_info.module_size = 100;
  134. rtStream_t stream = nullptr;
  135. callback(stream, &rt_kernel_info);
  136. return RT_ERROR_NONE;
  137. }
  138. rtError_t rtMemAdvise(void *ptr, uint64_t size, uint32_t advise) { return RT_ERROR_NONE; }
  139. /// @ingroup rt_kernel
  140. /// @brief start fusion kernels.
  141. /// @param [in] stream stream for fusion kernels
  142. /// @return RT_ERROR_NONE for ok, errno for failed
  143. rtError_t rtKernelFusionStart(rtStream_t stream) { return RT_ERROR_NONE; }
  144. /// @ingroup rt_kernel
  145. /// @brief end fusion kernels.
  146. /// @param [in] stream stream for fusion kernels
  147. /// @return RT_ERROR_NONE for ok, errno for failed
  148. rtError_t rtKernelFusionEnd(rtStream_t stream) { return RT_ERROR_NONE; }
  149. rtError_t rtMemGetInfo(size_t *free, size_t *total) {
  150. *free = 512UL * 1024UL * 1024UL;
  151. *total = 1024UL * 1024UL * 1024UL;
  152. return RT_ERROR_NONE;
  153. }
  154. rtError_t rtMemGetInfoEx(rtMemInfoType_t memInfoType, size_t *free, size_t *total) {
  155. *free = 512UL * 1024UL * 1024UL;
  156. *total = 1024UL * 1024UL * 1024UL;
  157. return RT_ERROR_NONE;
  158. }
  159. rtError_t rtMemAllocManaged(void **ptr, uint64_t size, uint32_t flag) { return RT_ERROR_NONE; }
  160. rtError_t rtMemFreeManaged(void *ptr) { return RT_ERROR_NONE; }
  161. rtError_t rtMetadataRegister(void *handle, const char *meta_data) { return RT_ERROR_NONE; }
  162. rtError_t rtSetTaskGenCallback(rtTaskGenCallback callback) { return RT_ERROR_NONE; }
  163. rtError_t rtModelCreate(rtModel_t *model, uint32_t flag) {
  164. *model = new uint32_t;
  165. return RT_ERROR_NONE;
  166. }
  167. rtError_t rtModelDestroy(rtModel_t model) {
  168. uint32_t *stub = static_cast<uint32_t *>(model);
  169. delete stub;
  170. return RT_ERROR_NONE;
  171. }
  172. rtError_t rtModelBindStream(rtModel_t model, rtStream_t stream, uint32_t flag) { return RT_ERROR_NONE; }
  173. rtError_t rtModelUnbindStream(rtModel_t model, rtStream_t stream) { return RT_ERROR_NONE; }
  174. rtError_t rtModelExecute(rtModel_t model, rtStream_t stream, uint32_t flag) { return RT_ERROR_NONE; }
  175. rtError_t rtGetFunctionByName(const char *stub_name, void **stub_func) {
  176. *(char **)stub_func = "func";
  177. return RT_ERROR_NONE;
  178. }
  179. rtError_t rtGetAddrByFun(const void *stubFunc, void **addr) {
  180. *(char **)addr = "dev_func";
  181. return RT_ERROR_NONE;
  182. }
  183. rtError_t rtQueryFunctionRegistered(const char *stub_name) { return RT_ERROR_NONE; }
  184. rtError_t rtCtxCreate(rtContext_t *ctx, uint32_t flags, int32_t device) { return RT_ERROR_NONE; }
  185. rtError_t rtKernelLaunchEx(void *args, uint32_t args_size, uint32_t flags, rtStream_t stream_) { return RT_ERROR_NONE; }
  186. rtError_t rtCpuKernelLaunch(const void *so_name, const void *kernel_name, uint32_t block_dim, const void *args,
  187. uint32_t args_size, rtSmDesc_t *sm_desc, rtStream_t stream) {
  188. return RT_ERROR_NONE;
  189. }
  190. rtError_t rtModelGetTaskId(void *handle, uint32_t *task_id, uint32_t *stream_id) {
  191. *task_id = 0;
  192. *stream_id = 0;
  193. return RT_ERROR_NONE;
  194. }
  195. rtError_t rtEndGraph(rtModel_t model, rtStream_t stream) { return RT_ERROR_NONE; }
  196. rtError_t rtEndGraphEx(rtModel_t model, rtStream_t stream, uint32_t flags)
  197. {
  198. return RT_ERROR_NONE;
  199. }
  200. rtError_t rtProfilerStop(uint64_t profConfig, int32_t numsDev, uint32_t *deviceList) {
  201. return RT_ERROR_NONE;
  202. }
  203. rtError_t rtSetDvfsProfile(DvfsProfileMode mode) { return RT_ERROR_NONE; }
  204. rtError_t rtUnsetDvfsProfile() { return RT_ERROR_NONE; }
  205. rtError_t rtGetDvfsProfile(DvfsProfileMode *pmode) { return RT_ERROR_NONE; }
  206. rtError_t rtCtxDestroy(rtContext_t ctx) { return RT_ERROR_NONE; }
  207. rtError_t rtProfilerInit(const char *prof_dir, const char *address, const char *job_ctx) { return RT_ERROR_NONE; }
  208. rtError_t rtProfilerStart(uint64_t profConfig, int32_t numsDev, uint32_t *deviceList) {
  209. return RT_ERROR_NONE;
  210. }
  211. rtError_t rtLabelCreate(rtLabel_t *label) {
  212. *label = new uint64_t;
  213. return RT_ERROR_NONE;
  214. }
  215. rtError_t rtLabelCreateEx(rtLabel_t *label, rtStream_t stream) {
  216. *label = new uint64_t;
  217. return RT_ERROR_NONE;
  218. }
  219. rtError_t rtLabelCreateV2(rtLabel_t *label, rtModel_t model) {
  220. *label = new uint64_t;
  221. return RT_ERROR_NONE;
  222. }
  223. rtError_t rtLabelCreateExV2(rtLabel_t *label, rtModel_t model, rtStream_t stream) {
  224. *label = new uint64_t;
  225. return RT_ERROR_NONE;
  226. }
  227. rtError_t rtLabelListCpy(rtLabel_t *label, uint32_t labelNumber, void *dst, uint32_t dstMax) {
  228. return RT_ERROR_NONE;
  229. }
  230. rtError_t rtLabelDestroy(rtLabel_t label) {
  231. uint64_t *stub = static_cast<uint64_t *>(label);
  232. delete stub;
  233. return RT_ERROR_NONE;
  234. }
  235. rtError_t rtLabelSet(rtLabel_t label, rtStream_t stream) { return RT_ERROR_NONE; }
  236. rtError_t rtLabelSwitch(void *ptr, rtCondition_t condition, uint32_t value, rtLabel_t true_label, rtStream_t stream) {
  237. return RT_ERROR_NONE;
  238. }
  239. rtError_t rtLabelSwitchByIndex(void *ptr, uint32_t max, void *labelInfoPtr, rtStream_t stream) {
  240. return RT_ERROR_NONE;
  241. }
  242. rtError_t rtLabelGoto(rtLabel_t label, rtStream_t stream) { return RT_ERROR_NONE; }
  243. rtError_t rtLabelGotoEx(rtLabel_t label, rtStream_t stream) {
  244. return RT_ERROR_NONE;
  245. }
  246. rtError_t rtInvalidCache(void *base, size_t len) {
  247. return RT_ERROR_NONE;
  248. }
  249. rtError_t rtModelLoadComplete(rtModel_t model) { return RT_ERROR_NONE; }
  250. rtError_t rtStreamCreateWithFlags(rtStream_t *stream, int32_t priority, uint32_t flags) {
  251. *stream = new uint32_t;
  252. return RT_ERROR_NONE;
  253. }
  254. rtError_t rtFlushCache(void *base, size_t len) {
  255. return RT_ERROR_NONE;
  256. }
  257. rtError_t rtProfilerTrace(uint64_t id, bool notify, uint32_t flags, rtStream_t stream_) { return RT_ERROR_NONE; }
  258. rtError_t rtProfilerTraceEx(uint64_t id, uint64_t modelId, uint16_t tagId, rtStream_t stream) { return RT_ERROR_NONE; }
  259. rtError_t rtMemSetRC(const void *dev_ptr, uint64_t size, uint32_t read_count) { return RT_ERROR_NONE; }
  260. rtError_t rtStreamSwitch(void *ptr, rtCondition_t condition, int64_t value, rtStream_t true_stream, rtStream_t stream) {
  261. return RT_ERROR_NONE;
  262. }
  263. rtError_t rtStreamSwitchEx(void *ptr, rtCondition_t condition, void *value_ptr, rtStream_t true_stream,
  264. rtStream_t stream, rtSwitchDataType_t data_type) {
  265. return RT_ERROR_NONE;
  266. }
  267. rtError_t rtStreamActive(rtStream_t active_stream, rtStream_t stream) { return RT_ERROR_NONE; }
  268. rtError_t rtEventReset(rtEvent_t event, rtStream_t stream) { return RT_ERROR_NONE; }
  269. rtError_t rtGetDevice(int32_t *device) { return RT_ERROR_NONE; }
  270. rtError_t rtDatadumpInfoLoad(const void *dump_info, uint32_t length) { return RT_ERROR_NONE; }
  271. rtError_t rtKernelLaunchWithFlag(const void *stub_func, uint32_t block_dim, void *args, uint32_t args_size,
  272. rtSmDesc_t *sm_desc, rtStream_t stream_, uint32_t flags) {
  273. return RT_ERROR_NONE;
  274. }
  275. rtError_t rtCpuKernelLaunchWithFlag(const void *so_name, const void *kernel_name, uint32_t core_dim, const void *args,
  276. uint32_t args_size, rtL2Ctrl_t *l2ctrl, rtStream_t stream_, uint32_t flags) {
  277. return RT_ERROR_NONE;
  278. }
  279. rtError_t rtModelGetId(rtModel_t model, uint32_t *modelId)
  280. {
  281. return RT_ERROR_NONE;
  282. }
  283. rtError_t rtModelBindQueue(rtModel_t model, uint32_t queueId, rtModelQueueFlag_t flag)
  284. {
  285. return RT_ERROR_NONE;
  286. }
  287. rtError_t rtSetSocVersion(const char *version)
  288. {
  289. return RT_ERROR_NONE;
  290. }
  291. rtError_t rtGetSocVersion(char *version, const uint32_t maxLen)
  292. {
  293. return RT_ERROR_NONE;
  294. }
  295. rtError_t rtGetAiCoreCount(uint32_t *aiCoreCnt)
  296. {
  297. return RT_ERROR_NONE;
  298. }
  299. rtError_t rtSetTaskFailCallback(rtTaskFailCallback callback)
  300. {
  301. return RT_ERROR_NONE;
  302. }
  303. rtError_t rtMallocHostSharedMemory(rtMallocHostSharedMemoryIn *in,
  304. rtMallocHostSharedMemoryOut *out)
  305. {
  306. out->ptr = new uint8_t[in->size];
  307. out->devPtr = new uint8_t[in->size];
  308. return RT_ERROR_NONE;
  309. }
  310. rtError_t rtFreeHostSharedMemory(rtFreeHostSharedMemoryIn *in)
  311. {
  312. delete[] (uint8_t*)in->ptr;
  313. delete[] (uint8_t*)in->devPtr;
  314. return RT_ERROR_NONE;
  315. }
  316. rtError_t rtGetAicpuDeploy(rtAicpuDeployType_t *deplyType)
  317. {
  318. return RT_ERROR_NONE;
  319. }
  320. rtError_t rtDebugRegister(rtModel_t model, uint32_t flag, const void *addr, uint32_t *streamId, uint32_t *taskId)
  321. {
  322. return RT_ERROR_NONE;
  323. }
  324. rtError_t rtDebugUnRegister(rtModel_t model)
  325. {
  326. return RT_ERROR_NONE;
  327. }
  328. rtError_t rtDumpAddrSet(rtModel_t model, void *addr, uint32_t dumpSize, uint32_t flag)
  329. {
  330. return RT_ERROR_NONE;
  331. }
  332. rtError_t rtSetCtxINFMode(bool mode)
  333. {
  334. return RT_ERROR_NONE;
  335. }
  336. rtError_t rtGetRtCapability(rtFeatureType_t featureType, int32_t featureInfo, int64_t *value)
  337. {
  338. return RT_ERROR_NONE;
  339. }
  340. rtError_t rtGetMaxStreamAndTask(uint32_t streamType, uint32_t *maxStrCount, uint32_t *maxTaskCount)
  341. {
  342. *maxStrCount = 1024;
  343. *maxTaskCount = 1024;
  344. return RT_ERROR_NONE;
  345. }
  346. rtError_t rtModelExit(rtModel_t model, rtStream_t stream)
  347. {
  348. return RT_ERROR_NONE;
  349. }
  350. rtError_t rtGetTaskIdAndStreamID(uint32_t *taskId, uint32_t *streamId)
  351. {
  352. return RT_ERROR_NONE;
  353. }
  354. rtError_t rtDebugRegisterForStream(rtStream_t stream, uint32_t flag, const void *addr, uint32_t *streamId, uint32_t *taskId) {
  355. return RT_ERROR_NONE;
  356. }
  357. rtError_t rtDebugUnRegisterForStream(rtStream_t stream) {
  358. return RT_ERROR_NONE;
  359. }
  360. rtError_t rtFftsTaskLaunch(rtFftsTaskInfo_t *fftsTaskInfo, rtStream_t stream) {
  361. return RT_ERROR_NONE;
  362. }
  363. #ifdef __cplusplus
  364. }
  365. #endif

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