csu3_am335x/EVSE/GPL/cmake-3.11.0/Tests/Cuda/WithC/cuda.cu

#include <cuda.h>

#include <iostream>

extern "C" int use_cuda(void)
{
  int nDevices = 0;
  cudaError_t err = cudaGetDeviceCount(&nDevices);
  if (err != cudaSuccess) {
    std::cerr << "Failed to retrieve the number of CUDA enabled devices"
              << std::endl;
    return 1;
  }
  std::cout << "Found " << nDevices << " CUDA enabled devices" << std::endl;
  return 0;
}