diff options
| -rwxr-xr-x | src/Core/regularisers_GPU/TV_FGP_GPU_core.cu | 2 |
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; } |