Searched refs:GetSingleKernelRuntime (Results 1 – 6 of 6) sorted by relevance
146 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in StartKernelRT()249 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in AllocateMemory()257 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in RunOpAllocateMemory()264 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in RunOpGenKernelEvent()271 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in RunOpClearMemory()404 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in CompileGraphImpl()607 … device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice, device_id_); in UpdateOutputTensors()631 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in Execute()722 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in DumpDataEnabledIteration()729 …auto runtime_instance = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(kGPUDevice… in SyncStream()
39 KernelRuntime *GetSingleKernelRuntime(const std::string &device_name, uint32_t device_id);
71 KernelRuntime *KernelRuntimeManager::GetSingleKernelRuntime(const std::string &device_name, uint32_… in GetSingleKernelRuntime() function in mindspore::device::KernelRuntimeManager
835 …auto runtime_instance = KernelRuntimeManager::Instance().GetSingleKernelRuntime(kAscendDevice, dev… in InitCtrlInputs()
33 …auto kernel_runtime = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(device_targe… in ConvertPhysicalDeviceId()
155 …auto runtime_instance = KernelRuntimeManager::Instance().GetSingleKernelRuntime(kAscendDevice, dev… in AllocateWorkspace()