CUDAUtilities/bin/cudaComputeCapabilities.cpp

0001 // C/C++ standard headers
0002 #include <cstdlib>
0003 #include <iomanip>
0004 #include <iostream>
0005
0006 // CUDA headers
0007 #include <cuda_runtime.h>
0008
0009 // CMSSW headers
0010 #include "isCudaDeviceSupported.h"
0011
0012 int main() {
0013   int devices = 0;
0014   cudaError_t status = cudaGetDeviceCount(&devices);
0015   if (status != cudaSuccess) {
0016     std::cerr << "cudaComputeCapabilities: " << cudaGetErrorString(status) << std::endl;
0017     return EXIT_FAILURE;
0018   }
0019
0020   for (int i = 0; i < devices; ++i) {
0021     cudaDeviceProp properties;
0022     cudaGetDeviceProperties(&properties, i);
0023     std::cout << std::setw(4) << i << "    " << std::setw(2) << properties.major << "." << properties.minor << "    "
0024               << properties.name;
0025     if (not isCudaDeviceSupported(i)) {
0026       std::cout << " (unsupported)";
0027     }
0028     std::cout << std::endl;
0029   }
0030
0031   return EXIT_SUCCESS;
0032 }