Mercurial > hg > Gears > Gears

changeset 412:409e6b5fb775

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add cuMemFree
author Tatsuki IHA <innparusu@cr.ie.u-ryukyu.ac.jp>
date Thu, 14 Sep 2017 19:11:25 +0900
parents 0eba9a04633f
children 497b154141de
files src/parallel_execution/cuda.c
diffstat 1 files changed, 2 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- 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) {