1 files changed, 2 insertions, 1 deletions
diff --git a/src/Core/regularisers_GPU/TV_ROF_GPU_core.cu b/src/Core/regularisers_GPU/TV_ROF_GPU_core.cu
index c6c6e88..e9e6d9b 100755
--- a/src/Core/regularisers_GPU/TV_ROF_GPU_core.cu
+++ b/src/Core/regularisers_GPU/TV_ROF_GPU_core.cu
@@ -480,7 +480,8 @@ extern "C" int TV_ROF_GPU_main(float* Input, float* Output, float *infovector, f
         CHECK(cudaFree(d_D2));
         CHECK(cudaFree(lambdaPar_d));
 
-	      infovector[0] = (float)(n);  /*iterations number (if stopped earlier based on tolerance)*/
+	infovector[0] = (float)(n);  /*iterations number (if stopped earlier based on tolerance)*/
         infovector[1] = re;  /* reached tolerance */
+        cudaDeviceSynchronize();
         return 0;
 }