diff options
Diffstat (limited to 'src/Core/regularisers_GPU/TV_PD_GPU_core.cu')
-rw-r--r-- | src/Core/regularisers_GPU/TV_PD_GPU_core.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/Core/regularisers_GPU/TV_PD_GPU_core.cu b/src/Core/regularisers_GPU/TV_PD_GPU_core.cu index 01e0ab5..3fcd1e8 100644 --- a/src/Core/regularisers_GPU/TV_PD_GPU_core.cu +++ b/src/Core/regularisers_GPU/TV_PD_GPU_core.cu @@ -515,5 +515,6 @@ extern "C" int TV_PD_GPU_main(float *Input, float *Output, float *infovector, fl /*adding info into info_vector */ infovector[0] = (float)(i); /*iterations number (if stopped earlier based on tolerance)*/ infovector[1] = re; /* reached tolerance */ + cudaDeviceSynchronize(); return 0; } |