cudareset for fgp_tv

1 files changed, 1 insertions, 1 deletions

diff --git a/src/Core/regularisers_GPU/TV_FGP_GPU_core.cu b/src/Core/regularisers_GPU/TV_FGP_GPU_core.cu
index ce2548f..cb8ba37 100755
--- a/src/Core/regularisers_GPU/TV_FGP_GPU_core.cu
+++ b/src/Core/regularisers_GPU/TV_FGP_GPU_core.cu
@@ -593,9 +593,9 @@ extern "C" int TV_FGP_GPU_main(float *Input, float *Output, float *infovector, f
             cudaFree(R2);
             cudaFree(R3);
     }
-    //cudaDeviceReset();
     /*adding info into info_vector */
     infovector[0] = (float)(i);  /*iterations number (if stopped earlier based on tolerance)*/
     infovector[1] = re;  /* reached tolerance */
+    cudaDeviceReset();
     return 0;
 }