elpa_index_gpu.cu 183 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
extern "C" {
  int gpu_count() {
    int count;
    cudaError_t cuerr = cudaGetDeviceCount(&count);
    if (cuerr != cudaSuccess) {
      count = -1000;
    }
    return count;
  }
}