Searched defs:numGPUs (Results 1 – 1 of 1) sorted by relevance
648 auto numGPUs = at::cuda::getNumGPUs(); in getCUDADeviceForRank() local1183 auto numGPUs = c10::cuda::device_count(); in barrier() local