1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
// Just a little test file to make sure that the CUDA library works
#include <c10/cuda/CUDAException.h>
#include <c10/cuda/impl/CUDATest.h>
#include <cuda_runtime.h>
namespace c10 {
namespace cuda {
namespace impl {
bool has_cuda_gpu() {
int count;
C10_CUDA_CHECK(cudaGetDeviceCount(&count));
return count != 0;
}
int c10_cuda_test() {
int r = 0;
if (has_cuda_gpu()) {
C10_CUDA_CHECK(cudaGetDevice(&r));
}
return r;
}
// This function is not exported
int c10_cuda_private_test() {
return 2;
}
}}} // namespace c10::cuda::impl
|
