Merge pull request #8 from kookaburra88/master

Update pdflow_cudalib.cu
MarianoJT88 · May 20, 2019 · 2d1e60c · 2d1e60c
2 parents 4b02fcf + cd516a6
commit 2d1e60c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pdflow_cudalib.cu b/pdflow_cudalib.cu
@@ -279,6 +279,10 @@ __host__ void CSF_cuda::freeDeviceMemory()
     cudaFree(puu_l_dev); cudaFree(puv_l_dev);
     cudaFree(pvu_l_dev); cudaFree(pvv_l_dev);
     cudaFree(pwu_l_dev); cudaFree(pwv_l_dev);
+
+	cudaFree(colour_wf_dev);
+	cudaFree(depth_wf_dev);
+
 
     cudaFree(dx_dev); cudaFree(dy_dev); cudaFree(dz_dev);
 }