[ref] set device namespace with CMake

ginkgo-project · Aug 23, 2024 · 114bf3e · 114bf3e
1 parent e567dd2
commit 114bf3e
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 16 deletions.
diff --git a/reference/CMakeLists.txt b/reference/CMakeLists.txt
@@ -66,6 +66,7 @@ target_sources(ginkgo_reference
     stop/residual_norm_kernels.cpp)
 
 target_link_libraries(ginkgo_reference PUBLIC ginkgo_device)
+target_compile_definitions(ginkgo_reference PRIVATE GKO_COMPILING_REFERENCE GKO_DEVICE_NAMESPACE=reference)
 ginkgo_compile_features(ginkgo_reference)
 ginkgo_default_includes(ginkgo_reference)
 ginkgo_install_library(ginkgo_reference)

diff --git a/reference/base/batch_multi_vector_kernels.cpp b/reference/base/batch_multi_vector_kernels.cpp
@@ -10,10 +10,6 @@
 #include <ginkgo/core/base/math.hpp>
 #include <ginkgo/core/base/range_accessors.hpp>
 
-
-#define GKO_DEVICE_NAMESPACE reference
-
-
 #include "core/base/batch_struct.hpp"
 #include "reference/base/batch_multi_vector_kernels.hpp"
 #include "reference/base/batch_struct.hpp"

diff --git a/reference/matrix/batch_csr_kernels.cpp b/reference/matrix/batch_csr_kernels.cpp
@@ -9,10 +9,6 @@
 #include <ginkgo/core/base/batch_multi_vector.hpp>
 #include <ginkgo/core/matrix/batch_csr.hpp>
 
-
-#define GKO_DEVICE_NAMESPACE reference
-
-
 #include "core/base/batch_struct.hpp"
 #include "core/matrix/batch_struct.hpp"
 #include "reference/base/batch_struct.hpp"

diff --git a/reference/matrix/batch_dense_kernels.cpp b/reference/matrix/batch_dense_kernels.cpp
@@ -9,10 +9,6 @@
 #include <ginkgo/core/base/batch_multi_vector.hpp>
 #include <ginkgo/core/matrix/batch_dense.hpp>
 
-
-#define GKO_DEVICE_NAMESPACE reference
-
-
 #include "core/base/batch_struct.hpp"
 #include "core/matrix/batch_struct.hpp"
 #include "reference/base/batch_struct.hpp"

diff --git a/reference/matrix/batch_ell_kernels.cpp b/reference/matrix/batch_ell_kernels.cpp
@@ -9,10 +9,6 @@
 #include <ginkgo/core/base/batch_multi_vector.hpp>
 #include <ginkgo/core/matrix/batch_ell.hpp>
 
-
-#define GKO_DEVICE_NAMESPACE reference
-
-
 #include "core/base/batch_struct.hpp"
 #include "core/matrix/batch_struct.hpp"
 #include "reference/base/batch_struct.hpp"