Prevent memory leak (yaks warning from dolfinx)

niravshah241 · Aug 30, 2024 · 95f47db · 95f47db
1 parent 7c2cc5b
commit 95f47db
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -5,9 +5,6 @@ FROM dolfinx/dolfinx:v0.7.3
 RUN mkdir -p /home/app
 WORKDIR /home/app
 
-# Copy package files
-COPY demo . 
-
 # Install MDFEniCSx
 RUN pip install git+https://github.com/niravshah241/MDFEniCSx.git
 

diff --git a/demo/0_fundamental_deformation/0_fundamentals.py b/demo/0_fundamental_deformation/0_fundamentals.py
@@ -91,6 +91,9 @@ def bc_side(x):
 ksp.getPC().setFactorSolverType("mumps")
 ksp.setFromOptions()
 ksp.solve(F, uh.vector)
+ksp.destroy()
+A.destroy()
+F.destroy()
 uh.x.scatter_forward()
 
 # uh now corresponds to the displacement of each node.

diff --git a/mdfenicsx/mesh_motion_classes.py b/mdfenicsx/mesh_motion_classes.py
@@ -109,6 +109,9 @@ def solve(self):
         ksp.getPC().setFactorSolverType("mumps")
         ksp.setFromOptions()
         ksp.solve(F, uh.vector)
+        ksp.destroy()
+        A.destroy()
+        F.destroy()
         uh.x.scatter_forward()
         return uh
 
@@ -166,7 +169,7 @@ def sigma(self, u_func):
                                                             ("DG", 0))
         # Function for Lam\'e parameter lambda = E / (2 * (1 + \nu))
         lambda_ = dolfinx.fem.Function(material_function_space)
-        # Function for Lam\'e parameter mu = (E * \nu) / ((1-2*\nu) * (1+\nu))
+        # Function for Lam\'e parameter mu_ = (E * \nu) / ((1-2*\nu) * (1+\nu))
         mu_ = dolfinx.fem.Function(material_function_space)
         # Assemble discontinuous material property functions
         for i in range(len(self._young_modulus_list)):