Searched refs:func_stub (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 45 auto func_stub = KernelManager::GenFuncStub(*kernel_pack_, false, &blockdim); in Launch() local
46 if (func_stub == 0) { in Launch()
62 const void *stubFunc = reinterpret_cast<void *>(func_stub); in Launch()
149 …auto func_stub = KernelManager::GenFuncStub(*kernel_pack_, false, &block_dim, dynamic_flag, &handl… in GenDynamicKernel() local
151 if (func_stub != 1) { in GenDynamicKernel()
157 if (func_stub == 0) { in GenDynamicKernel()
160 const void *stub_func_ptr = reinterpret_cast<void *>(func_stub); in GenDynamicKernel()
|
| D | tbe_utils.cc | 320 uintptr_t func_stub = ++kernel_stub_gen_; in GenFuncStub() local
322 …rtFunctionRegister(module, reinterpret_cast<void *>(func_stub), func_name.c_str(), func_name.c_str… in GenFuncStub()
327 info_table_[func_name] = std::make_shared<KernelMetaInfo>(KernelMetaInfo{func_stub, *block_dim}); in GenFuncStub()
328 return func_stub; in GenFuncStub()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_mod.cc | 67 auto func_stub = KernelManager::GenFuncStub(*kernel_pack_, false, &block_dim); in Launch() local
68 if (func_stub == 0) { in Launch()
86 …if (RT_ERROR_NONE != rtKernelLaunch(reinterpret_cast<void *>(func_stub), block_dim, runtime_args.d… in Launch()
123 auto func_stub = KernelManager::GenFuncStub(*kernel_pack_, false, &block_dim); in GenTask() local
124 if (func_stub == 0) { in GenTask()
|