From 821a868a635e5de63d96524b9f2fb7b56770e549 Mon Sep 17 00:00:00 2001
From: Kris Thielemans <k.thielemans@ucl.ac.uk>
Date: Wed, 10 Jul 2024 16:17:37 +0100
Subject: [PATCH] use CUDARDP if it exists

---
 petric.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/petric.py b/petric.py
index 56ffcb2..208ff78 100755
--- 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)