Lines Matching defs:cuda_gpu_id
107 const int cuda_gpu_id = GpuIdManager::TfToCudaGpuId(tf_gpu_id).value();
108 device_prop_ = &Eigen::m_deviceProperties[cuda_gpu_id];
682 int32 cuda_gpu_id;
683 if (!strings::safe_strto32(cuda_gpu_id_str, &cuda_gpu_id)) {
688 if (cuda_gpu_id < 0 || cuda_gpu_id >= gpu_manager->VisibleDeviceCount()) {
690 "'visible_device_list' listed an invalid GPU id '", cuda_gpu_id,
694 visible_gpu_order->push_back(CudaGpuId(cuda_gpu_id));
780 // 'cuda_gpu_id', when that virtual device is the only virtual device being
783 CudaGpuId cuda_gpu_id,
788 GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie();
791 cuda_gpu_id.value());
884 for (CudaGpuId cuda_gpu_id : valid_cuda_gpu_ids) {
885 err = cudaSetDevice(cuda_gpu_id.value());
887 return errors::Internal("cudaSetDevice() on GPU:", cuda_gpu_id.value(),
893 "CUDA runtime implicit initialization on GPU:", cuda_gpu_id.value(),
946 const CudaGpuId cuda_gpu_id = valid_cuda_gpu_ids[i];
951 gpu_options, cuda_gpu_id, &single_virtual_device_memory_limit));
963 GpuIdManager::InsertTfCudaGpuIdPair(tf_gpu_id, cuda_gpu_id);
988 static string GetShortDeviceDescription(CudaGpuId cuda_gpu_id,
997 return strings::StrCat("device: ", cuda_gpu_id.value(),
1014 CudaGpuId cuda_gpu_id = GpuIdManager::TfToCudaGpuId(tf_gpu_id);
1019 GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie();
1023 << GetShortDeviceDescription(cuda_gpu_id, desc) << ")";
1027 GetShortDeviceDescription(cuda_gpu_id, desc),
1086 CudaGpuId cuda_gpu_id = GpuIdManager::TfToCudaGpuId(tf_gpu_id);
1091 GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie();
1101 LOG(INFO) << "Could not identify NUMA node of CUDA gpu id " << cuda_gpu_id
1115 if (imap.directed_links.find({cuda_gpu_id, cuda_gpu_dst}) !=
1130 if (cuda_gpu_id == cuda_gpu_dst) {
1139 VLOG(1) << "GPUDevice CudaGpuId " << cuda_gpu_id << " TfGpuId " << tf_gpu_id
1300 const CudaGpuId cuda_gpu_id = visible_gpu_order[i];
1303 if (visible_gpu_initialized_[cuda_gpu_id.value()]) {
1307 visible_gpu_initialized_[cuda_gpu_id.value()] = true;
1310 auto executor = GpuIdUtil::ExecutorForCudaGpuId(gpu_manager, cuda_gpu_id);
