CMAKE: add check for ZFP_CUDA

ornladios · Oct 29, 2021 · d0fbb7f · d0fbb7f
1 parent 35c3f19
commit d0fbb7f
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 6 deletions.
diff --git a/cmake/DetectOptions.cmake b/cmake/DetectOptions.cmake
@@ -95,6 +95,30 @@ elseif(ADIOS2_USE_ZFP)
 endif()
 if(ZFP_FOUND)
   set(ADIOS2_HAVE_ZFP TRUE)
+  set(ADIOS2_HAVE_ZFP_CUDA ${ZFP_CUDA})
+
+  # Older versions of ZFP
+  if(NOT ADIOS2_HAVE_ZFP_CUDA)
+    get_target_property(ZFP_INCLUDE_DIRECTORIES zfp::zfp INTERFACE_INCLUDE_DIRECTORIES)
+    set(CMAKE_REQUIRED_INCLUDES ${ZFP_INCLUDE_DIRECTORIES})
+    set(CMAKE_REQUIRED_LIBRARIES zfp::zfp)
+    include(CheckCSourceRuns)
+    check_c_source_runs("
+    #include <zfp.h>
+
+    int main()
+    {
+      zfp_stream* stream = zfp_stream_open(NULL);
+      return !zfp_stream_set_execution(stream, zfp_exec_cuda);
+    }"
+    ADIOS2_HAVE_ZFP_CUDA)
+    unset(CMAKE_REQUIRED_INCLUDES)
+    unset(CMAKE_REQUIRED_LIBRARIES)
+  endif()
+
+  if(ADIOS2_HAVE_ZFP_CUDA)
+    add_compile_definitions(ADIOS2_HAVE_ZFP_CUDA)
+  endif()
 endif()
 
 # SZ

diff --git a/source/adios2/common/ADIOSTypes.h b/source/adios2/common/ADIOSTypes.h
@@ -312,7 +312,9 @@ constexpr char precision[] = "precision";
 
 namespace value
 {
+#ifdef ADIOS2_HAVE_ZFP_CUDA
 constexpr char backend_cuda[] = "cuda";
+#endif
 constexpr char backend_omp[] = "omp";
 constexpr char backend_serial[] = "serial";
 }

diff --git a/source/adios2/operator/compress/CompressZFP.cpp b/source/adios2/operator/compress/CompressZFP.cpp
@@ -13,7 +13,11 @@
 
 /* ZFP will default to SERIAL if CUDA is not available */
 #ifndef ZFP_DEFAULT_EXECUTION_POLICY
+#ifdef ADIOS2_HAVE_ZFP_CUDA
 #define ZFP_DEFAULT_EXECUTION_POLICY zfp_exec_cuda
+#else
+#define ZFP_DEFAULT_EXECUTION_POLICY zfp_exec_serial
+#endif
 #endif
 
 namespace adios2
@@ -282,19 +286,21 @@ zfp_stream *CompressZFP::GetZFPStream(const Dims &dimensions, DataType type,
         auto policy = ZFP_DEFAULT_EXECUTION_POLICY;
         const auto backend = itBackend->second;
 
-        if (backend == "cuda")
+        if (backend == "serial")
         {
-            policy = zfp_exec_cuda;
+            policy = zfp_exec_serial;
+            isSerial = true;
         }
         else if (backend == "omp")
         {
             policy = zfp_exec_omp;
         }
-        else if (backend == "serial")
+#ifdef ADIOS2_HAVE_ZFP_CUDA
+        else if (backend == "cuda")
         {
-            policy = zfp_exec_serial;
-            isSerial = true;
+            policy = zfp_exec_cuda;
         }
+#endif
 
         zfp_stream_set_execution(stream, policy);
     }

diff --git a/testing/adios2/engine/bp/operations/CMakeLists.txt b/testing/adios2/engine/bp/operations/CMakeLists.txt
@@ -30,7 +30,7 @@ if(ADIOS2_HAVE_ZFP)
       )
   endforeach()
 
-  if(ADIOS2_HAVE_CUDA)
+  if(ADIOS2_HAVE_CUDA AND ADIOS2_HAVE_ZFP_CUDA)
     enable_language(CUDA)
 
     gtest_add_tests_helper(WriteReadZfpCuda MPI_ALLOW BP Engine.BP. .BP4