1 #include <iostream>
2 
3 #include <cuda.h>
4 
use_cuda(void)5 extern "C" int use_cuda(void)
6 {
7   int nDevices = 0;
8   cudaError_t err = cudaGetDeviceCount(&nDevices);
9   if (err != cudaSuccess) {
10     std::cerr << "Failed to retrieve the number of CUDA enabled devices"
11               << std::endl;
12     return 1;
13   }
14   std::cout << "Found " << nDevices << " CUDA enabled devices" << std::endl;
15   return 0;
16 }
17