From 1a3d1c19cfff0a93db6028192a44aec461eb66f6 Mon Sep 17 00:00:00 2001
From: Bartek Kryza <bkryza@gmail.com>
Date: Wed, 1 May 2024 21:27:33 +0200
Subject: [PATCH] Fixed cmake CUDA target defines

---
 CMakeLists.txt       | 3 ---
 tests/CMakeLists.txt | 5 +++--
 tests/test_cases.cc  | 2 ++
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 6941ccbe..17124374 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -162,9 +162,6 @@ option(ENABLE_CUDA_TEST_CASES "" OFF)
 if(ENABLE_CUDA_TEST_CASES)
     include(CheckLanguage)
     check_language(CUDA)
-    if(DEFINED CMAKE_CUDA_COMPILER)
-        set(ENABLE_CUDA_TEST_CASES ON)
-    endif(DEFINED CMAKE_CUDA_COMPILER)
 endif(ENABLE_CUDA_TEST_CASES)
 
 if(BUILD_TESTS)
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index f2c94d8e..3ce15f5f 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -32,7 +32,7 @@ else()
 endif(ENABLE_CXX_MODULES_TEST_CASES)
 
 if(NOT ENABLE_CUDA_TEST_CASES)
-    message(STATUS "Enabling CUDA test cases")
+    message(STATUS "Disabling CUDA test cases")
     foreach(CUDA_TC ${TEST_CASES_REQUIRING_CUDA})
         list(FILTER TEST_CASE_SOURCES
                 EXCLUDE
@@ -112,7 +112,8 @@ foreach(TEST_NAME ${TEST_NAMES})
             $<IF:${ENABLE_CXX_STD_20_TEST_CASES},cxx_std_20,cxx_std_17>)
     target_compile_definitions(${TEST_NAME} PRIVATE
             $<$<EQUAL:${ENABLE_CXX_STD_20_TEST_CASES},1>:ENABLE_CXX_STD_20_TEST_CASES>
-            $<$<BOOL:${ENABLE_CXX_MODULES_TEST_CASES}>:ENABLE_CXX_MODULES_TEST_CASES>)
+            $<$<BOOL:${ENABLE_CXX_MODULES_TEST_CASES}>:ENABLE_CXX_MODULES_TEST_CASES>
+            $<$<BOOL:${ENABLE_CUDA_TEST_CASES}>:ENABLE_CUDA_TEST_CASES>)
     target_compile_options(${TEST_NAME} PRIVATE
             $<$<COMPILE_LANGUAGE:CXX>:
             $<$<OR:$<CXX_COMPILER_ID:Clang>,$<CXX_COMPILER_ID:GNU>>:
diff --git a/tests/test_cases.cc b/tests/test_cases.cc
index 080737f3..5797bf67 100644
--- a/tests/test_cases.cc
+++ b/tests/test_cases.cc
@@ -475,9 +475,11 @@ using namespace clanguml::test::matchers;
 #include "t20046/test_case.h"
 #include "t20047/test_case.h"
 #include "t20048/test_case.h"
+#if defined(ENABLE_CUDA_TEST_CASES)
 #include "t20049/test_case.h"
 #include "t20050/test_case.h"
 #include "t20051/test_case.h"
+#endif
 
 ///
 /// Package diagram tests