From d4d12945f5396b10259fdeeb2b09f99b0e2c6afd Mon Sep 17 00:00:00 2001
From: algol <dkazanc@hotmail.com>
Date: Thu, 22 Feb 2018 12:34:39 +0000
Subject: GPU test fixed, cpu vs gpu test added for ROF-TV

---
 Wrappers/Python/src/gpu_regularizers.pyx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'Wrappers/Python/src')

diff --git a/Wrappers/Python/src/gpu_regularizers.pyx b/Wrappers/Python/src/gpu_regularizers.pyx
index fcb91cc..c724471 100644
--- a/Wrappers/Python/src/gpu_regularizers.pyx
+++ b/Wrappers/Python/src/gpu_regularizers.pyx
@@ -345,7 +345,7 @@ def ROFTV2D(np.ndarray[np.float32_t, ndim=2, mode="c"] inputData,
     # Running CUDA code here    
     TV_ROF_GPU_kernel(            
             &inputData[0,0], &B[0,0], 
-                       dims[0], dims[1], 0, 
+                       dims[0], dims[1], 1, 
                        iterations , 
                        time_marching_parameter, 
                        regularization_parameter);   
-- 
cgit v1.2.3