use CUDARDP if it exists

SyneRBI · Jul 10, 2024 · 821a868 · 821a868
1 parent 8437da1
commit 821a868
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/petric.py b/petric.py
@@ -136,7 +136,11 @@ def construct_RDP(penalty_strength, initial_image, kappa, max_scaling=1e-3):
     initial_image: used to determine a smoothing factor (epsilon).
     kappa: used to pass voxel-dependent weights.
     """
-    prior = STIR.RelativeDifferencePrior()
+    try:
+        prior = STIR.CudaRelativeDifferencePrior()
+    except:
+        prior = STIR.RelativeDifferencePrior()
+
     # need to make it differentiable
     epsilon = initial_image.max() * max_scaling
     prior.set_epsilon(epsilon)