diff options
Diffstat (limited to 'src/Core/regularisers_GPU/PatchSelect_GPU_core.cu')
-rw-r--r-- | src/Core/regularisers_GPU/PatchSelect_GPU_core.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/Core/regularisers_GPU/PatchSelect_GPU_core.cu b/src/Core/regularisers_GPU/PatchSelect_GPU_core.cu index fb6fa95..cde252e 100644 --- a/src/Core/regularisers_GPU/PatchSelect_GPU_core.cu +++ b/src/Core/regularisers_GPU/PatchSelect_GPU_core.cu @@ -505,6 +505,6 @@ extern "C" int PatchSelect_GPU_main(float *A, unsigned short *H_i, unsigned shor cudaFree(H_j_d); cudaFree(Weights_d); cudaFree(Eucl_Vec_d); - cudaDeviceReset(); + cudaDeviceSynchronize(); return 0; } |