diff options
Diffstat (limited to 'src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu')
| -rw-r--r-- | src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu b/src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu index 89fca06..12473c0 100644 --- a/src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu +++ b/src/Core/regularisers_GPU/dTV_FGP_GPU_core.cu @@ -753,5 +753,6 @@ extern "C" int dTV_FGP_GPU_main(float *Input, float *InputRef, float *Output, 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; } |