# HG changeset patch # User Tatsuki IHA # Date 1505383885 -32400 # Node ID 409e6b5fb775d9453441cc9a5c7917ea79d70687 # Parent 0eba9a04633fbf6ca6178006a80f5df65afb82e1 Add cuMemFree diff -r 0eba9a04633f -r 409e6b5fb775 src/parallel_execution/cuda.c --- a/src/parallel_execution/cuda.c Thu Sep 14 18:26:52 2017 +0900 +++ b/src/parallel_execution/cuda.c Thu Sep 14 19:11:25 2017 +0900 @@ -127,6 +127,8 @@ // wait for stream checkCudaErrors(cuCtxSynchronize()); + cuMemFree(devA); + cuMemFree(devB); } void cudaShutdown( struct CUDAWorker *worker) {