Tests/Cuda/WithC/cuda.cu

   1 #include <iostream>
   2
   3 #include <cuda.h>
   4
   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 }