Home
last modified time | relevance | path

Searched refs:runtime_instance (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc146 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in StartKernelRT() local
147 MS_EXCEPTION_IF_NULL(runtime_instance); in StartKernelRT()
148 if (!runtime_instance->Init()) { in StartKernelRT()
249 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in AllocateMemory() local
250 MS_EXCEPTION_IF_NULL(runtime_instance); in AllocateMemory()
251 runtime_instance->AssignMemory(*kernel_graph); in AllocateMemory()
257 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in RunOpAllocateMemory() local
258 MS_EXCEPTION_IF_NULL(runtime_instance); in RunOpAllocateMemory()
259 runtime_instance->RunOpAssignMemory(input_tensors, *kernel_graph); in RunOpAllocateMemory()
264 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in RunOpGenKernelEvent() local
[all …]
Dascend_session.cc514 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in LoadInputData() local
515 MS_EXCEPTION_IF_NULL(runtime_instance); in LoadInputData()
516 auto compute_stream = runtime_instance->compute_stream(); in LoadInputData()
517 auto model_stream = runtime_instance->GetModelStream(kernel_graph->graph_id()); in LoadInputData()
518 auto memcpy_event = runtime_instance->CreateDeviceEvent(); in LoadInputData()
629 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in CompileGraphImpl() local
630 MS_EXCEPTION_IF_NULL(runtime_instance); in CompileGraphImpl()
631 uint64_t mem_size = runtime_instance->GetAvailableMemMaxSize(); in CompileGraphImpl()
740 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in CompileChildGraph() local
741 MS_EXCEPTION_IF_NULL(runtime_instance); in CompileChildGraph()
[all …]
Dsession_basic.cc2593 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(devcie_target, d… in InitPsWorker() local
2594 MS_EXCEPTION_IF_NULL(runtime_instance); in InitPsWorker()
2595 auto context = runtime_instance->context(); in InitPsWorker()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc80 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in SyncMemory() local
81 MS_EXCEPTION_IF_NULL(runtime_instance); in SyncMemory()
82 runtime_instance->SetContext(); in SyncMemory()
91 …auto ret = runtime_instance->MemcpyAsync(dst, src, size, static_cast<int32_t>(RT_MEMCPY_HOST_TO_DE… in SyncMemory()
106 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in DataSync() local
107 MS_EXCEPTION_IF_NULL(runtime_instance); in DataSync()
108 …auto ret = runtime_instance->MemcpyAsync(dst, src, src_size, static_cast<int32_t>(RT_MEMCPY_DEVICE… in DataSync()
172 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, d… in SyncStream() local
173 MS_EXCEPTION_IF_NULL(runtime_instance); in SyncStream()
174 auto ret = runtime_instance->SyncStream(); in SyncStream()
[all …]
Dascend_bucket.cc57 auto runtime_instance = device::KernelRuntimeManager::Instance().GetCurrentKernelRuntime(); in AllocateAllReduceAddr() local
58 MS_EXCEPTION_IF_NULL(runtime_instance); in AllocateAllReduceAddr()
60 ar_input_addr_ = runtime_instance->MallocCommunicationMemFromMemPool(total_size); in AllocateAllReduceAddr()
61 ar_output_addr_ = runtime_instance->MallocCommunicationMemFromMemPool(total_size); in AllocateAllReduceAddr()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dai_core_dynamic_kernel.cc155 …auto runtime_instance = KernelRuntimeManager::Instance().GetSingleKernelRuntime(kAscendDevice, dev… in AllocateWorkspace() local
156 MS_EXCEPTION_IF_NULL(runtime_instance); in AllocateWorkspace()
161 auto device_ptr = runtime_instance->MallocMem(MemType::kDynamicMem, size, device_address_ptr); in AllocateWorkspace()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc1301 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(device_name, dev… in InitHccl()
1302 MS_EXCEPTION_IF_NULL(runtime_instance); in InitHccl()
1304 runtime_instance->PreInit(); in InitHccl()
1307 if (!runtime_instance->Init()) { in InitHccl()
1397 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetKernelRuntime(device_name, dev… in StartUpProfiling() local
1398 MS_EXCEPTION_IF_NULL(runtime_instance); in StartUpProfiling()
1399 runtime_instance->PreInit(); in StartUpProfiling()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc835 …auto runtime_instance = KernelRuntimeManager::Instance().GetSingleKernelRuntime(kAscendDevice, dev… in InitCtrlInputs() local
836 …if (runtime_instance->MallocMem(kStaticMem, LongToSize(tensor->data().nbytes()), device_address) =… in InitCtrlInputs()