Searched refs:gpuGuard (Results 1 – 4 of 4) sorted by relevance
44 c10::cuda::CUDAGuard gpuGuard(storage.device()); in gds_load_storage() local60 c10::cuda::CUDAGuard gpuGuard(storage.device()); in gds_save_storage() local
437 at::DeviceGuard gpuGuard(device); in newLikeFlat() local451 at::DeviceGuard gpuGuard(t.device()); in newLikeFlat() local
1175 at::cuda::OptionalCUDAGuard gpuGuard; in abortCommsFromMap() local1182 gpuGuard.set_index(deviceIndex); in abortCommsFromMap()2180 at::cuda::OptionalCUDAGuard gpuGuard(device); in checkForNCCLErrorsInternal() local2724 at::cuda::OptionalCUDAGuard gpuGuard(device); in checkForNCCLErrorsInternal() local2907 at::cuda::OptionalCUDAGuard gpuGuard(device); in checkForNCCLErrorsInternal() local3175 at::cuda::OptionalCUDAGuard gpuGuard(device); in checkForNCCLErrorsInternal() local
672 c10::cuda::OptionalCUDAGuard gpuGuard; in runHealthCheck() local688 gpuGuard.set_index(device.index()); in runHealthCheck()