refinement instead of nvars for PETSc

Author: pancetta

pySDC/implementations/problem_classes/HeatEquation_2D_PETSc_forced.py

@@ -39,26 +39,28 @@ def __init__(self, problem_params, dtype_u=petsc_data, dtype_f=rhs_imex_petsc_da
         if 'sol_maxiter' not in problem_params:
             problem_params['sol_maxiter'] = None
 
-        essential_keys = ['nvars', 'nu', 'freq', 'comm']
+        essential_keys = ['cnvars', 'nu', 'freq', 'comm', 'refine']
         for key in essential_keys:
             if key not in problem_params:
                 msg = 'need %s to instantiate problem, only got %s' % (key, str(problem_params.keys()))
                 raise ParameterError(msg)
 
         # make sure parameters have the correct form
-        if len(problem_params['nvars']) != 2:
-            raise ProblemError('this is a 2d example, got %s' % problem_params['nvars'])
+        if len(problem_params['cnvars']) != 2:
+            raise ProblemError('this is a 2d example, got %s' % problem_params['cnvars'])
 
         # create DMDA object which will be used for all grid operations
-        da = PETSc.DMDA().create([problem_params['nvars'][0], problem_params['nvars'][1]], stencil_width=1,
+        da = PETSc.DMDA().create([problem_params['cnvars'][0], problem_params['cnvars'][1]], stencil_width=1,
                                  comm=problem_params['comm'])
+        for _ in range(problem_params['refine']):
+            da = da.refine()
 
         # invoke super init, passing number of dofs, dtype_u and dtype_f
         super(heat2d_petsc_forced, self).__init__(init=da, dtype_u=dtype_u, dtype_f=dtype_f, params=problem_params)
 
         # compute dx, dy and get local ranges
-        self.dx = 1.0 / (self.params.nvars[0] - 1)
-        self.dy = 1.0 / (self.params.nvars[1] - 1)
+        self.dx = 1.0 / (self.init.getSizes()[0] - 1)
+        self.dy = 1.0 / (self.init.getSizes()[1] - 1)
         (self.xs, self.xe), (self.ys, self.ye) = self.init.getRanges()
 
         # compute discretization matrix A and identity

pySDC/playgrounds/PETSc/playground_data.py

@@ -12,7 +12,10 @@ def main():
     dx = 1.0/(n - 1)
     dy = dx
     comm= PETSc.COMM_WORLD
-    da = PETSc.DMDA().create([n, n], dof=2, stencil_width=1, comm=comm)
+    da = PETSc.DMDA().create([n, n], dof=1, stencil_width=1, comm=comm)
+    dar = da.refine()
+    print(dar.getSizes())
+    exit()
 
     rank = PETSc.COMM_WORLD.getRank()
     # comm=

pySDC/projects/TOMS/pySDC_with_PETSc.py

@@ -60,7 +60,8 @@ def main():
     problem_params = dict()
     problem_params['nu'] = 1.0  # diffusion coefficient
     problem_params['freq'] = 2  # frequency for the test value
-    problem_params['nvars'] = [(129, 129), (65, 65)]  # number of degrees of freedom for each level
+    problem_params['cnvars'] = [(257, 257)]  # number of degrees of freedom on coarse level
+    problem_params['refine'] = [1, 0]  # number of refinements
     problem_params['comm'] = space_comm  # pass space-communicator to problem class
     problem_params['sol_tol'] = 1E-10  # set tolerance to PETSc' linear solver
 
@@ -76,7 +77,7 @@ def main():
 
     # initialize controller parameters
     controller_params = dict()
-    controller_params['logger_level'] = 30 if space_rank == 0 else 99  # set level depending on rank
+    controller_params['logger_level'] = 20 if space_rank == 0 else 99  # set level depending on rank
     controller_params['dump_setup'] = False
 
     # fill description dictionary for easy step instantiation

pySDC/tutorial/step_7/C_pySDC_with_PETSc.py

@@ -55,7 +55,8 @@ def main():
     problem_params = dict()
     problem_params['nu'] = 1.0  # diffusion coefficient
     problem_params['freq'] = 2  # frequency for the test value
-    problem_params['nvars'] = [(129, 129), (65, 65)]  # number of degrees of freedom for each level
+    problem_params['cnvars'] = [(65, 65)]  # number of degrees of freedom for the coarsest level
+    problem_params['refine'] = [1, 0]  # number of refinements
     problem_params['comm'] = space_comm  # pass space-communicator to problem class
     problem_params['sol_tol'] = 1E-12  # set tolerance to PETSc' linear solver