mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
4a019047ad
It is time to enable nested namespaces in the code. Pull Request resolved: https://github.com/pytorch/pytorch/pull/118506 Approved by: https://github.com/albanD
31 lines
541 B
C++
31 lines
541 B
C++
// 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::cuda::impl {
|
|
|
|
bool has_cuda_gpu() {
|
|
int count = 0;
|
|
C10_CUDA_IGNORE_ERROR(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
|