#include <iostream> | |
#include <cuda.h> | |
#include <cuda_runtime.h> | |
int __host__ cu2_sq_func(int x) | |
{ | |
cudaError_t err; | |
int nDevices = 0; | |
err = cudaGetDeviceCount(&nDevices); | |
if (err != cudaSuccess) { | |
std::cerr << "nDevices: " << nDevices << std::endl; | |
std::cerr << "err: " << err << std::endl; | |
return 1; | |
} | |
return x * x; | |
} |