Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 12:49:28

0001 #include <cstdlib>
0002 #include <iostream>
0003 
0004 #include <cuda_runtime.h>
0005 
0006 #include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h"
0007 
0008 namespace cms::cudatest {
0009   bool testDevices() {
0010     int devices = 0;
0011     auto status = cudaGetDeviceCount(&devices);
0012     if (status != cudaSuccess) {
0013       std::cerr << "Failed to initialise the CUDA runtime, the test will be skipped."
0014                 << "\n";
0015       return false;
0016     }
0017     if (devices == 0) {
0018       std::cerr << "No CUDA devices available, the test will be skipped."
0019                 << "\n";
0020       return false;
0021     }
0022     return true;
0023   }
0024 
0025   void requireDevices() {
0026     if (not testDevices()) {
0027       exit(EXIT_SUCCESS);
0028     }
0029   }
0030 }  // namespace cms::cudatest