From e0eb65274b88625dee7c00758a957cf908f84b40 Mon Sep 17 00:00:00 2001 From: dkazanc Date: Fri, 28 Feb 2020 15:39:09 +0000 Subject: cudareset for fgp_tv --- src/Core/regularisers_GPU/TV_FGP_GPU_core.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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; } -- cgit v1.2.3