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)
 
@@ -207,6 +208,20 @@ else()
   message(STATUS "Parallelproj projector support disabled or not available.")
 endif()
 
+if(NOT DISABLE_STIR_CUDA)
+  find_package(CUDAToolkit)
+  if (CUDAToolkit_FOUND)
+    set(CMAKE_CUDA_ARCHITECTURES "all")
+    find_package(CUDA REQUIRED)
+    include_directories("${CUDA_INCLUDE_DIRS}")
+    set(STIR_WITH_STIR_CUDA ON)
+    if(STIR_WITH_STIR_CUDA)
+      enable_language(CUDA)
+      message(STATUS "STIR CUDA support enabled. Using CUDA version ${CUDAToolkit_VERSION}.")
+    endif()
+  endif()
+endif()
+
 #### enable support for ctest
 ENABLE_TESTING()
 

src/cmake/STIRConfig.cmake.in

@@ -143,6 +143,11 @@ if(@STIR_WITH_NiftyPET_PROJECTOR@)
   set(STIR_WITH_NiftyPET_PROJECTOR TRUE)
 endif()
 
+if(@STIR_WITH_STIR_CUDA@)
+  find_package(CUDAToolkit REQUIRED)
+  set(STIR_WITH_STIR_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_STIR_CUDA
+
 #cmakedefine STIR_WITH_Parallelproj_PROJECTOR
 #cmakedefine parallelproj_built_with_CUDA
 

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

@@ -0,0 +1,36 @@
+#ifndef CUDA_RELATIVE_DIFFERENCE_PRIOR_CLASS_H
+#define CUDA_RELATIVE_DIFFERENCE_PRIOR_CLASS_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 // CUDA_RELATIVE_DIFFERENCE_PRIOR_CLASS_H

src/recon_buildblock/CMakeLists.txt

@@ -126,6 +126,11 @@ if (STIR_WITH_Parallelproj_PROJECTOR)
   )
 endif()
 
+if (STIR_WITH_STIR_CUDA)
+  list(APPEND ${dir_LIB_SOURCES}
+    CUDA_stir/CudaRelativeDifferencePrior.cu
+  )
+endif()
 
 include(stir_lib_target)
 
@@ -155,3 +160,7 @@ if (STIR_WITH_Parallelproj_PROJECTOR)
     target_link_libraries(recon_buildblock PUBLIC parallelproj::parallelproj_cuda)
   endif()
 endif()
+
+if (STIR_WITH_STIR_CUDA)
+  target_link_libraries(recon_buildblock PUBLIC CUDA::cudart)
+endif()