From 8b822d8e802c98f87a0e32c85f1243578d5555e7 Mon Sep 17 00:00:00 2001
From: Thomas Padioleau <thomas.padioleau@cea.fr>
Date: Tue, 10 Sep 2024 18:25:40 +0200
Subject: [PATCH] Fix unit test headers installation

---
 batched/CMakeLists.txt                  | 2 --
 batched/dense/unit_test/CMakeLists.txt  | 3 +++
 batched/sparse/unit_test/CMakeLists.txt | 3 +++
 common/CMakeLists.txt                   | 1 -
 common/unit_test/CMakeLists.txt         | 3 +++
 ode/CMakeLists.txt                      | 1 -
 ode/unit_test/CMakeLists.txt            | 3 +++
 7 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/batched/CMakeLists.txt b/batched/CMakeLists.txt
index 3103dfa8a0..544bd713b3 100644
--- a/batched/CMakeLists.txt
+++ b/batched/CMakeLists.txt
@@ -4,11 +4,9 @@ LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_BINARY_DIR}/batched/eti)
 
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/dense/src)
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/dense/impl)
-LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/dense/unit_test)
 
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/sparse/src)
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/sparse/impl)
-LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/batched/sparse/unit_test)
 
 IF (NOT KokkosKernels_ENABLE_COMPONENT_BLAS)
   LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/blas/impl)
diff --git a/batched/dense/unit_test/CMakeLists.txt b/batched/dense/unit_test/CMakeLists.txt
index f970e81e66..e9ad450691 100644
--- a/batched/dense/unit_test/CMakeLists.txt
+++ b/batched/dense/unit_test/CMakeLists.txt
@@ -18,6 +18,9 @@ KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_C
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR}/batched/dense/impl)
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR}/batched/dense/impl)
 
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR})
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR})
+
 IF (NOT KokkosKernels_ENABLE_COMPONENT_BLAS)
   KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_BINARY_DIR}/blas/src)
   KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${PACKAGE_SOURCE_DIR}/blas/src)
diff --git a/batched/sparse/unit_test/CMakeLists.txt b/batched/sparse/unit_test/CMakeLists.txt
index 63c34906b3..7f0a6d7a4f 100644
--- a/batched/sparse/unit_test/CMakeLists.txt
+++ b/batched/sparse/unit_test/CMakeLists.txt
@@ -18,6 +18,9 @@ KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_C
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR}/batched/dense/impl)
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR}/batched/sparse/impl)
 
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR})
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR})
+
 IF (NOT KokkosKernels_ENABLE_COMPONENT_BLAS)
   MESSAGE("batched enabled and blas not enabled, we need to include some include directories manually!")
   KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_BINARY_DIR}/src/blas)
diff --git a/common/CMakeLists.txt b/common/CMakeLists.txt
index b065869296..3961dbff9a 100644
--- a/common/CMakeLists.txt
+++ b/common/CMakeLists.txt
@@ -1,4 +1,3 @@
 # Adding source directory to the build
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/common/src)
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/common/impl)
-LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/common/unit_test)
diff --git a/common/unit_test/CMakeLists.txt b/common/unit_test/CMakeLists.txt
index c0d8fc116f..c727a74e80 100644
--- a/common/unit_test/CMakeLists.txt
+++ b/common/unit_test/CMakeLists.txt
@@ -7,6 +7,9 @@
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_BINARY_DIR}/test_common)
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${PACKAGE_SOURCE_DIR}/test_common)
 
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR})
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR})
+
 #####################
 #                   #
 # Define unit-tests #
diff --git a/ode/CMakeLists.txt b/ode/CMakeLists.txt
index b9cf089445..409268e9b1 100644
--- a/ode/CMakeLists.txt
+++ b/ode/CMakeLists.txt
@@ -1,6 +1,5 @@
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/ode/src)
 LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/ode/impl)
-LIST(APPEND KK_INCLUDE_DIRS ${CMAKE_CURRENT_SOURCE_DIR}/ode/unit_test)
 
 IF (NOT KokkosKernels_ENABLE_COMPONENT_BATCHED)
   MESSAGE("blas enabled and batched not enabled, we need to include some headers manually!")
diff --git a/ode/unit_test/CMakeLists.txt b/ode/unit_test/CMakeLists.txt
index 90d6c45c49..27d407896d 100644
--- a/ode/unit_test/CMakeLists.txt
+++ b/ode/unit_test/CMakeLists.txt
@@ -18,6 +18,9 @@ KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_C
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR}/ode/impl)
 KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR}/ode/impl)
 
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(${CMAKE_CURRENT_BINARY_DIR})
+KOKKOSKERNELS_INCLUDE_DIRECTORIES(REQUIRED_DURING_INSTALLATION_TESTING ${CMAKE_CURRENT_SOURCE_DIR})
+
 #####################
 #                   #
 # Define unit-tests #