| #include <cuda.h> | |
| #include <cuda_runtime.h> | |
| #include <iostream> | |
| 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; | |
| } |