Searched refs:kernel_runtime (Results 1 – 6 of 6) sorted by relevance
100 std::shared_ptr<KernelRuntime> kernel_runtime; in GetKernelRuntime() local104 kernel_runtime = (creator_iter->second)(); in GetKernelRuntime()105 MS_EXCEPTION_IF_NULL(kernel_runtime); in GetKernelRuntime()106 kernel_runtime->set_device_id(device_id); in GetKernelRuntime()107 runtime_map_[runtime_key] = kernel_runtime; in GetKernelRuntime()112 return kernel_runtime.get(); in GetKernelRuntime()
2 … "kernel_info.cc" "executor/dynamic_kernel.cc" "executor/executor_callback.cc" "kernel_runtime.cc"
33 …auto kernel_runtime = device::KernelRuntimeManager::Instance().GetSingleKernelRuntime(device_targe… in ConvertPhysicalDeviceId() local34 MS_EXCEPTION_IF_NULL(kernel_runtime); in ConvertPhysicalDeviceId()35 return kernel_runtime->device_id(); in ConvertPhysicalDeviceId()
740 auto kernel_runtime = device::KernelRuntimeManager::Instance().GetCurrentKernelRuntime(); in CreateBucket() local741 MS_EXCEPTION_IF_NULL(kernel_runtime); in CreateBucket()742 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket()743 auto communication_stream = kernel_runtime->communication_stream(); in CreateBucket()
1581 …auto kernel_runtime = device::KernelRuntimeManager::Instance().GetKernelRuntime(kAscendDevice, dev… in DumpAllGraphs() local1582 MS_EXCEPTION_IF_NULL(kernel_runtime); in DumpAllGraphs()1945 auto kernel_runtime = device::KernelRuntimeManager::Instance().GetCurrentKernelRuntime(); in CreateBucket() local1946 MS_EXCEPTION_IF_NULL(kernel_runtime); in CreateBucket()1947 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket()1948 auto communication_stream = kernel_runtime->communication_stream(); in CreateBucket()
111 "../../../mindspore/ccsrc/runtime/device/kernel_runtime.cc"