summaryrefslogtreecommitdiffstats
path: root/Tests/CudaOnly/SeparateCompilation/main/main.cu
diff options
context:
space:
mode:
authorroot <raul@tambre.ee>2021-07-27 19:58:03 (GMT)
committerroot <raul@tambre.ee>2021-07-29 10:19:37 (GMT)
commit0b1cea66cd1f80458f0da579d0182d908874939d (patch)
tree182874aefd315752ab7a5e3c583a4a13d21e5ae4 /Tests/CudaOnly/SeparateCompilation/main/main.cu
parentf7cf69e34a1607e8ea2b6d10fef6a6058377c24e (diff)
downloadCMake-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.cu68
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;
+}