Searched refs:cudaSetDevice (Results 1 – 13 of 13) sorted by relevance
140 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()150 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
175 [&] { XLA_CUDA_WARN_IF_ERROR(cudaSetDevice(initial_cuda_device)); }); in CreateNcclClique()184 cudaSetDevice(local_participants[i].device_ordinal)); in CreateNcclClique()
43 #define cudaSetDevice hipSetDevice macro
48 #define cudaSetDevice hipSetDevice408 CUDA_RETURN_IF_ERROR(cudaSetDevice(devices[i])); in GetCommunicator()413 CUDA_RETURN_IF_ERROR(cudaSetDevice(saved_device)); in GetCommunicator()
446 cudaSetDevice(cuda_device_id); in CreateTRTNode()826 cudaSetDevice(old_cuda_device.value()); in ConvertAfterShapes()
1218 err = cudaSetDevice(platform_gpu_id.value()); in CreateDevices()1271 err = cudaSetDevice(original_device); in CreateDevices()
1062 auto err = cudaSetDevice(platform_gpu_id); in AllocateCalibrationResources()
255 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {257 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
256 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {258 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
265 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {267 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");
288 extern __host__ cudaError_t CUDARTAPI cudaSetDevice(int device) {290 static auto func_ptr = LoadSymbol<FuncPtr>("cudaSetDevice");