Add sub buffer release

pwollstadt · Feb 25, 2021 · 9980e17 · 9980e17
1 parent 7a1fbce
commit 9980e17
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/idtxl/estimators_opencl.py b/idtxl/estimators_opencl.py
@@ -437,6 +437,9 @@ def _estimate_single_run(self, var1, var2, n_chunks=1):
         d_distances.release()
         d_npointsrange_x.release()
         d_npointsrange_y.release()
+        d_var1.release()
+        d_var2.release()
+        d_vecradius.release()
 
         # Calculate and sum digammas
         if self.settings['local_values']:
@@ -771,6 +774,9 @@ def _estimate_single_run(self, var1, var2, conditional=None, n_chunks=1):
         d_npointsrange_x.release()
         d_npointsrange_y.release()
         d_npointsrange_z.release()
+        d_src.release()
+        d_cnd.release()
+        d_vecradius.release()
 
         # Calculate and sum digammas
         if self.settings['local_values']: