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