Searched refs:KernelRuntime (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime_manager.h | 31 using KernelRuntimeCreator = std::function<std::shared_ptr<KernelRuntime>()>;
37 KernelRuntime *GetKernelRuntime(const std::string &device_name, uint32_t device_id);
38 KernelRuntime *GetCurrentKernelRuntime();
39 KernelRuntime *GetSingleKernelRuntime(const std::string &device_name, uint32_t device_id);
49 std::map<std::string, std::shared_ptr<KernelRuntime> > runtime_map_;
|
| D | kernel_runtime.cc | 72 KernelRuntime::~KernelRuntime() { in ~KernelRuntime()
78 bool KernelRuntime::Load(const session::KernelGraph &, bool) { in Load()
83 bool KernelRuntime::LoadData(const session::KernelGraph &) { in LoadData()
88 bool KernelRuntime::NodeOutputDeviceAddressExist(const AnfNodePtr &kernel, size_t index) { in NodeOutputDeviceAddressExist()
98 void KernelRuntime::AssignMemory(const session::KernelGraph &graph) { in AssignMemory()
114 void KernelRuntime::RunOpGetCommunicationInputInfo(const AnfNodePtr &node, size_t *total_size, in RunOpGetCommunicationInputInfo()
144 void KernelRuntime::RunOpAssignCommunicationInput(const AnfNodePtr &node) const { in RunOpAssignCommunicationInput()
170 void KernelRuntime::RunOpGetCommunicationOutputInfo(const AnfNodePtr &node, size_t *total_size, in RunOpGetCommunicationOutputInfo()
198 void KernelRuntime::RunOpAssignCommunicationOutput(const AnfNodePtr &node) const { in RunOpAssignCommunicationOutput()
220 void KernelRuntime::RunOpMallocPre(const session::KernelGraph &graph, in RunOpMallocPre()
[all …]
|
| D | kernel_runtime_manager.cc | 71 KernelRuntime *KernelRuntimeManager::GetSingleKernelRuntime(const std::string &device_name, uint32_… in GetSingleKernelRuntime()
93 KernelRuntime *KernelRuntimeManager::GetKernelRuntime(const std::string &device_name, uint32_t devi… in GetKernelRuntime()
100 std::shared_ptr<KernelRuntime> kernel_runtime; in GetKernelRuntime()
115 KernelRuntime *KernelRuntimeManager::GetCurrentKernelRuntime() { in GetCurrentKernelRuntime()
|
| D | kernel_runtime.h | 51 class KernelRuntime {
53 KernelRuntime() = default;
54 virtual ~KernelRuntime();
198 using KernelRuntimePtr = std::shared_ptr<KernelRuntime>;
|
| D | device_address.h | 115 friend class KernelRuntime; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | hccl_dynamic_kernel.cc | 48 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in UpdateArgs()
86 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in StaticShapeExecute()
|
| D | ai_core_dynamic_kernel.cc | 113 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in UpdateArgs()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.h | 33 class CPUKernelRuntime : public KernelRuntime {
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.h | 39 class AscendKernelRuntime : public KernelRuntime {
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | memcpy_async.cc | 139 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.h | 39 class GPUKernelRuntime : public KernelRuntime {
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 117 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 210 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 279 device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &inputs, &workspaces, &outputs); in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger.cc | 213 return device::KernelRuntime::DumpDataEnabled(); in CheckDebuggerDumpEnabled()
1437 …if (device_target_ == kGPUDevice && (debugger_enabled_ || device::KernelRuntime::DumpDataEnabledIt… in UpdateStepNum()
1455 …if ((device_target_ == kGPUDevice) && (debugger_enabled_ || device::KernelRuntime::DumpDataEnabled… in ClearCurrentData()
|