compiler: Suitably destroy the submats inside the jacobianstruct

Author: ZoeLeibowitz

devito/petsc/iet/type_builder.py

@@ -119,7 +119,7 @@ def _extend_build(self, base_dict):
 
         base_dict['jacctx'] = JacobianStruct(
             name=sreg.make_name(prefix=objs['ljacctx'].name),
-            fields=objs['ljacctx'].fields, no_of_submats=2
+            fields=objs['ljacctx'].fields, no_of_submats=len(targets)*len(targets)
         )
 
         for sm in submatrices:

devito/petsc/types/object.py

@@ -279,27 +279,12 @@ def __init__(self, name='jctx', pname='JacobianCtx', fields=None,
 
     @property
     def _C_free(self):
-        # from IPython import embed; embed()
         submats = [i for i in self.fields if isinstance(i, PointerMat)]
         submats = submats[0]
-        # from IPython import embed; embed()
         from devito.symbolics import FieldFromComposite
         destroy_call = [petsc_call('MatDestroy', [Byref(FieldFromComposite(submats.indexed[i], self.function))]) for i in range(self._no_of_submats)]
         destroy_call.append(petsc_call('PetscFree', [Byref(FieldFromComposite(submats.base, self.function))]))
         return destroy_call
-        # return petsc_call('PetscFree', [Byref(self.function)])
-    
-
-    # @property
-    # def _C_free(self):
-
-    #     submats = [i for i in self.fields if isinstance(i, PointerMat)]
-    #     destroy_calls = [
-    #         petsc_call('MatDestroy', [Byref(self.indexify().subs({self.dim: i}))])
-    #         for i in range(self._no_of_submats)
-    #     ]
-    #     destroy_calls.append(petsc_call('PetscFree', [self.function]))
-    #     return destroy_calls
 
 
 class SubMatrixStruct(PETScStruct):