Searched defs: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
1175 at::cuda::OptionalCUDAGuard gpuGuard; in abortCommsFromMap() local2180 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
437 at::DeviceGuard gpuGuard(device); in newLikeFlat() local451 at::DeviceGuard gpuGuard(t.device()); in newLikeFlat() local
672 c10::cuda::OptionalCUDAGuard gpuGuard; in runHealthCheck() local