CUDARelativeDifferencePrior

CMakeLists.txt

@@ -101,6 +101,7 @@ option(DISABLE_NLOHMANN_JSON "disable use of nlohmann JSON libraries" OFF)
 option(STIR_ENABLE_EXPERIMENTAL "disable use of STIR experimental code" OFF) # disable by default
 option(DISABLE_NiftyPET_PROJECTOR "disable use of NiftyPET projector" OFF)
 option(DISABLE_Parallelproj_PROJECTOR "disable use of Parallelproj projector" OFF)
+option(DISABLE_STIR_CUDA "disable use of STIR cuda" OFF)
 OPTION(DOWNLOAD_ZENODO_TEST_DATA "download zenodo data for tests" OFF)
 option(DISABLE_UPENN "disable use of UPENN filetypes" OFF)
 
@@ -186,6 +187,24 @@ else()
   message(STATUS "NiftyPET projector support disabled.")
 endif()
 
+if(NOT DISABLE_STIR_CUDA)
+  ## check 
+  find_package(CUDAToolkit REQUIRED)
+  find_package(CUDA REQUIRED)
+  if("${CMAKE_VERSION}" VERSION_LESS "3.23")
+    message(FATAL_ERROR "CMake 3.23 or newer is required to use CMAKE_CUDA_ARCHITECTURES set to 'all'")
+  else()
+    set(CMAKE_CUDA_ARCHITECTURES "all")
+  endif()
+  if (CUDAToolkit_FOUND)
+    if(STIR_WITH_CUDA)
+      enable_language(CUDA)
+      message(STATUS "STIR CUDA support enabled. Using CUDA version ${CUDAToolkit_VERSION}.")
+      set(STIR_WITH_CUDA ON)
+    endif()
+  endif()
+endif()
+
 # Parallelproj
 if(NOT DISABLE_Parallelproj_PROJECTOR)
     find_package(parallelproj 1.3.4 CONFIG)

src/CMakeLists.txt

@@ -244,6 +244,9 @@ add_library(stir_registries OBJECT ${STIR_REGISTRIES})
 # TODO, really should use stir_libs.cmake
 target_include_directories(stir_registries PRIVATE ${STIR_INCLUDE_DIR})
 target_include_directories(stir_registries PRIVATE ${Boost_INCLUDE_DIR})
+if (STIR_WITH_CUDA)
+  target_include_directories(stir_registries PRIVATE ${CUDA_INCLUDE_DIRS})
+endif()
 
 # go and look for CMakeLists.txt files in all those directories
 foreach(STIR_DIR ${STIR_DIRS} ${STIR_TEST_DIRS})

src/cmake/STIRConfig.cmake.in

@@ -143,6 +143,12 @@ if(@STIR_WITH_NiftyPET_PROJECTOR@)
   set(STIR_WITH_NiftyPET_PROJECTOR TRUE)
 endif()
 
+if(@STIR_WITH_CUDA@)
+  find_package(CUDAToolkit REQUIRED)
+  enable_language(CUDA)
+  set(STIR_WITH_CUDA TRUE)
+endif()
+
 if(@STIR_WITH_Parallelproj_PROJECTOR@)
   find_package(parallelproj REQUIRED CONFIG)
   set(STIR_WITH_Parallelproj_PROJECTOR TRUE)

src/cmake/STIRConfig.h.in

@@ -69,6 +69,8 @@ namespace stir {
 
 #cmakedefine STIR_WITH_NiftyPET_PROJECTOR
 
+#cmakedefine STIR_WITH_CUDA
+
 #cmakedefine STIR_WITH_Parallelproj_PROJECTOR
 #cmakedefine parallelproj_built_with_CUDA
 

src/include/stir/recon_buildblock/CUDA/CudaRelativeDifferencePrior.h

@@ -0,0 +1,36 @@
+#ifndef __stir_recon_buildblock_CudaRelativeDifferencePrior_h__
+#define __stir_recon_buildblock_CudaRelativeDifferencePrior_h__
+
+#include "stir/recon_buildblock/RelativeDifferencePrior.h"
+#include "stir/DiscretisedDensity.h"
+#include <cuda_runtime.h>
+#include "stir/Succeeded.h"
+
+START_NAMESPACE_STIR
+
+template <typename elemT>
+class CudaRelativeDifferencePrior : public RelativeDifferencePrior<elemT> {
+    public:
+        using RelativeDifferencePrior<elemT>::RelativeDifferencePrior;
+        // Name which will be used when parsing a GeneralisedPrior object
+        static const char* const registered_name;
+
+        // Constructors
+        CudaRelativeDifferencePrior();
+        CudaRelativeDifferencePrior(const bool only_2D, float penalization_factor, float gamma, float epsilon);
+
+        // Overridden methods
+        double compute_value(const DiscretisedDensity<3, elemT>& current_image_estimate) override;
+        void compute_gradient(DiscretisedDensity<3, elemT>& prior_gradient, const DiscretisedDensity<3, elemT>& current_image_estimate) override;
+
+        // New methods
+        Succeeded set_up_cuda(shared_ptr<DiscretisedDensity<3, elemT>> const& target_sptr);
+    protected:
+        int z_dim, y_dim, x_dim;
+        dim3 grid_dim, block_dim;
+        bool cuda_set_up = false;
+};
+
+END_NAMESPACE_STIR
+
+#endif // __stir_recon_buildblock_CudaRelativeDifferencePrior_h__

src/recon_buildblock/CMakeLists.txt

@@ -126,10 +126,14 @@ if (STIR_WITH_Parallelproj_PROJECTOR)
   )
 endif()
 
+if (STIR_WITH_CUDA)
+  list(APPEND ${dir_LIB_SOURCES}
+    CUDA/CudaRelativeDifferencePrior.cu
+  )
+endif()
 
 include(stir_lib_target)
 
-
 if (STIR_MPI)
   target_link_libraries(recon_buildblock PUBLIC ${MPI_CXX_LIBRARIES})
 endif()
@@ -155,3 +159,7 @@ if (STIR_WITH_Parallelproj_PROJECTOR)
     target_link_libraries(recon_buildblock PUBLIC parallelproj::parallelproj_cuda)
   endif()
 endif()
+
+if (STIR_WITH_CUDA)
+  target_link_libraries(recon_buildblock PUBLIC CUDA::cudart)
+endif()