misc: Create getargs Op inside function

Author: ZoeLeibowitz

devito/petsc/initialize.py

@@ -55,14 +55,12 @@ def PetscInitialize(clargs=sys.argv):
         _petsc_initialized = True
 
 
-with switchconfig(language='petsc'):
-    op_get_args = Operator(
-        [PetscEq(dummy, GetArgs(dummy))],
-        name='kernel_get_args', opt='noop'
-    )
-
-
 def PetscGetArgs():
+    with switchconfig(language='petsc'):
+        op_get_args = Operator(
+            [PetscEq(dummy, GetArgs(dummy))],
+            name='kernel_get_args', opt='noop'
+        )
     argc_ptr = c_int()
     argv_ptr = (POINTER(c_char_p))()
     op_get_args.apply(argc=byref(argc_ptr), argv=byref(argv_ptr))

examples/petsc/init_test.py

@@ -1,8 +1,10 @@
 import os
-from devito.petsc.initialize import PetscInitialize
+from devito.petsc.initialize import PetscInitialize, PetscGetArgs
 from devito import configuration
 configuration['compiler'] = 'custom'
 os.environ['CC'] = 'mpicc'
 
 PetscInitialize()
 print("helloworld")
+
+PetscGetArgs()