Searched refs:gpu_manager (Results 1 – 4 of 4) sorted by relevance
34 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs() local35 if (gpu_manager != nullptr) { in GetNumAvailableGPUs()36 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs()38 auto exec_status = gpu_manager->ExecutorForDevice(i); in GetNumAvailableGPUs()
771 se::Platform* gpu_manager = GPUMachineManager(); in ParseVisibleDeviceList() local776 visible_gpu_order->resize(gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()793 platform_gpu_id >= gpu_manager->VisibleDeviceCount()) { in ParseVisibleDeviceList()797 gpu_manager->VisibleDeviceCount()); in ParseVisibleDeviceList()982 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices() local983 if (gpu_manager == nullptr) { in CreateDevices()987 if (gpu_manager->VisibleDeviceCount() <= 0) { in CreateDevices()1078 GetInterconnectMaps(visible_gpu_order, gpu_manager, &interconnect_maps)); in CreateDevices()1257 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps) { in GetInterconnectMaps() argument1259 auto access_map = GetPeerAccessMap(gpu_manager, visible_gpu_order); in GetInterconnectMaps()[all …]
34 se::Platform* gpu_manager, PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId() argument35 return gpu_manager->ExecutorForDevice(platform_gpu_id.value()); in ExecutorForPlatformGpuId()
282 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps);