add use_cuda option to Example_compute_gradient

Author: joeljonsson

examples/Example_compute_gradient.cpp

@@ -59,10 +59,29 @@ int main(int argc, char **argv) {
     ParticleData<float> output;
     std::vector<float> deltas = {options.dy, options.dx, options.dz};
 
-    if(options.sobel) {
-        APRNumerics::gradient_magnitude_sobel(apr, parts, output, deltas);
-    } else {
-        APRNumerics::gradient_magnitude_cfd(apr, parts, output, deltas);
+    bool done = false;
+
+    if(options.use_cuda) {
+#ifdef APR_USE_CUDA
+        auto access = apr.gpuAPRHelper();
+        auto tree_access = apr.gpuTreeHelper();
+        if(options.sobel) {
+            APRNumericsGPU::gradient_magnitude_sobel(access, tree_access, parts.data, output.data, deltas);
+        } else {
+            APRNumericsGPU::gradient_magnitude_cfd(access, tree_access, parts.data, output.data, deltas);
+        }
+        done = true;
+#else
+        std::cout << "Option -use_cuda was given, but LibAPR was not built with CUDA enabled. Using CPU implementation." << std::endl;
+#endif
+    }
+
+    if(!done) {
+        if (options.sobel) {
+            APRNumerics::gradient_magnitude_sobel(apr, parts, output, deltas);
+        } else {
+            APRNumerics::gradient_magnitude_cfd(apr, parts, output, deltas);
+        }
     }
     timer.stop_timer();
 
@@ -144,6 +163,11 @@ cmdLineOptions read_command_line_options(int argc, char **argv){
         result.dz =  std::stof(get_command_option(argv, argv + argc, "-dz"));
     }
 
+    if(command_option_exists(argv, argv + argc, "-use_cuda"))
+    {
+        result.use_cuda = true;
+    }
+
 
     return result;
 

examples/Example_compute_gradient.hpp

@@ -17,6 +17,10 @@
 #include "numerics/APRNumerics.hpp"
 #include "numerics/APRReconstruction.hpp"
 
+#ifdef APR_USE_CUDA
+#include "numerics/APRNumericsGPU.hpp"
+#endif
+
 struct cmdLineOptions{
     std::string output = "";
     std::string stats = "";
@@ -26,6 +30,7 @@ struct cmdLineOptions{
     float dx = 1.0f;
     float dy = 1.0f;
     float dz = 1.0f;
+    bool use_cuda = false;
 };
 
 cmdLineOptions read_command_line_options(int argc, char **argv);