diff options
author | root <raul@tambre.ee> | 2021-07-27 19:58:03 (GMT) |
---|---|---|
committer | root <raul@tambre.ee> | 2021-07-29 10:19:37 (GMT) |
commit | 0b1cea66cd1f80458f0da579d0182d908874939d (patch) | |
tree | 182874aefd315752ab7a5e3c583a4a13d21e5ae4 /Tests/CudaOnly/SeparateCompilation/main/main.cu | |
parent | f7cf69e34a1607e8ea2b6d10fef6a6058377c24e (diff) | |
download | CMake-0b1cea66cd1f80458f0da579d0182d908874939d.zip CMake-0b1cea66cd1f80458f0da579d0182d908874939d.tar.gz CMake-0b1cea66cd1f80458f0da579d0182d908874939d.tar.bz2 |
CUDA/Clang: Fix separable compilation in non-root directories with Makefiles
Seems the relative paths were wrong basically all around such that only
compiling files in the top-level directory would work. I've modified
CudaOnly.SeparateCompilation to cover this.
Fixes #22482.
Diffstat (limited to 'Tests/CudaOnly/SeparateCompilation/main/main.cu')
-rw-r--r-- | Tests/CudaOnly/SeparateCompilation/main/main.cu | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/Tests/CudaOnly/SeparateCompilation/main/main.cu b/Tests/CudaOnly/SeparateCompilation/main/main.cu new file mode 100644 index 0000000..2b6e8f4 --- /dev/null +++ b/Tests/CudaOnly/SeparateCompilation/main/main.cu @@ -0,0 +1,68 @@ + +#include <iostream> + +#include "../file1.h" +#include "../file2.h" + +int file4_launch_kernel(int x); +int file5_launch_kernel(int x); + +int choose_cuda_device() +{ + int nDevices = 0; + cudaError_t err = cudaGetDeviceCount(&nDevices); + if (err != cudaSuccess) { + std::cerr << "Failed to retrieve the number of CUDA enabled devices" + << std::endl; + return 1; + } + for (int i = 0; i < nDevices; ++i) { + cudaDeviceProp prop; + cudaError_t err = cudaGetDeviceProperties(&prop, i); + if (err != cudaSuccess) { + std::cerr << "Could not retrieve properties from CUDA device " << i + << std::endl; + return 1; + } + if (prop.major >= 3) { + err = cudaSetDevice(i); + if (err != cudaSuccess) { + std::cout << "Could not select CUDA device " << i << std::endl; + } else { + return 0; + } + } + } + + std::cout << "Could not find a CUDA enabled card supporting compute >=3.0" + << std::endl; + + return 1; +} + +int main(int argc, char** argv) +{ + int ret = choose_cuda_device(); + if (ret) { + return 0; + } + + cudaError_t err; + file4_launch_kernel(42); + err = cudaGetLastError(); + if (err != cudaSuccess) { + std::cerr << "file4_launch_kernel: kernel launch failed: " + << cudaGetErrorString(err) << std::endl; + return 1; + } + + file5_launch_kernel(42); + err = cudaGetLastError(); + if (err != cudaSuccess) { + std::cerr << "file5_launch_kernel: kernel launch failed: " + << cudaGetErrorString(err) << std::endl; + return 1; + } + + return 0; +} |