Searched refs:cudaSetDevice (Results 1 – 8 of 8) sorted by relevance
140 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()150 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
80 cudaError_t CUDART_CB cudaSetDevice(int device) { in cudaSetDevice() function
319 CUDA_RETURN_IF_ERROR(cudaSetDevice(devices[i])); in GetCommunicator()324 CUDA_RETURN_IF_ERROR(cudaSetDevice(saved_device)); in GetCommunicator()
918 cudaSetDevice(cuda_device_id); in ConvertAfterShapes()948 cudaSetDevice(old_cuda_device); in ConvertAfterShapes()
1033 err = cudaSetDevice(platform_gpu_id.value()); in CreateDevices()1062 err = cudaSetDevice(original_device); in CreateDevices()
698 auto err = cudaSetDevice(platform_gpu_id); in AllocateCalibrationResources()
583 cudaSetDevice(mpi_global.device); in BackgroundThreadLoop()
584 cudaSetDevice(mpi_global.device); in BackgroundThreadLoop()