Transformation skeleton for ParallelRoutineDispatchTransformation

Author: reuterbal

transformations/tests/sources/projParallelRoutineDispatch/dispatch_routine.F90

@@ -0,0 +1,54 @@
+SUBROUTINE DISPATCH_ROUTINE(YDGEOMETRY, YDCPG_BNDS, YDCPG_OPTS, &
+    & YDCPG_PHY0, YDCPG_DYN0, YDMF_PHYS_SURF, YDVARS,       &
+    & YDMODEL)
+
+USE GEOMETRY_MOD       , ONLY : GEOMETRY
+USE CPG_TYPE_MOD       , ONLY : CPG_DYN_TYPE, &
+                                & CPG_PHY_TYPE
+USE CPG_OPTS_TYPE_MOD  , ONLY : CPG_BNDS_TYPE, CPG_OPTS_TYPE
+USE MF_PHYS_SURFACE_TYPE_MOD  &
+                        & , ONLY : MF_PHYS_SURF_TYPE
+USE FIELD_VARIABLES_MOD, ONLY : FIELD_VARIABLES
+USE TYPE_MODEL         , ONLY : MODEL
+
+USE PARKIND1           , ONLY : JPIM     ,JPRB
+USE YOMHOOK            , ONLY : LHOOK    ,DR_HOOK, JPHOOK
+
+IMPLICIT NONE
+
+TYPE(GEOMETRY),                 INTENT(IN)    :: YDGEOMETRY
+TYPE(CPG_BNDS_TYPE),            INTENT(IN)    :: YDCPG_BNDS
+TYPE(CPG_OPTS_TYPE),            INTENT(IN)    :: YDCPG_OPTS
+TYPE(CPG_PHY_TYPE),             INTENT(IN)    :: YDCPG_PHY0
+TYPE(CPG_DYN_TYPE),             INTENT(IN)    :: YDCPG_DYN0
+TYPE(MF_PHYS_SURF_TYPE),        INTENT(INOUT) :: YDMF_PHYS_SURF
+TYPE(FIELD_VARIABLES),          INTENT(INOUT) :: YDVARS
+TYPE(MODEL),                    INTENT(IN)    :: YDMODEL
+
+REAL(KIND=JPRB)     :: ZRDG_CVGQ (YDCPG_OPTS%KLON, 1:YDCPG_OPTS%KFLEVG)
+REAL(KIND=JPRB)     :: ZRDG_MU0LU (YDCPG_OPTS%KLON)
+REAL(KIND=JPRB)     :: ZRDG_MU0M (YDCPG_OPTS%KLON)
+REAL(KIND=JPRB)     :: ZRDG_MU0N (YDCPG_OPTS%KLON)
+REAL(KIND=JPRB)     :: ZRDG_MU0 (YDCPG_OPTS%KLON)
+
+INTEGER(KIND=JPIM) :: INSTEP_DEB, INSTEP_FIN
+INTEGER(KIND=JPIM) :: JLEV, JLON, JSPP
+
+REAL(KIND=JPHOOK) :: ZHOOK_HANDLE
+
+IF (LHOOK) CALL DR_HOOK('DISPATCH_ROUTINE',0,ZHOOK_HANDLE)
+
+INSTEP_DEB=1
+INSTEP_FIN=1
+!$ACDC PARALLEL,TARGET=OpenMP/OpenMPSingleColumn/OpenACCSingleColumn,NAME=CPPHINP {
+
+CALL CPPHINP(YDGEOMETRY, YDMODEL, YDCPG_BNDS%KIDIA, YDCPG_BNDS%KFDIA, YDVARS%GEOMETRY%GEMU%T0,                     &
+& YDVARS%GEOMETRY%GELAM%T0, YDVARS%U%T0, YDVARS%V%T0, YDVARS%Q%T0, YDVARS%Q%DL, YDVARS%Q%DM, YDVARS%CVGQ%DL,       &
+& YDVARS%CVGQ%DM, YDCPG_PHY0%XYB%RDELP, YDCPG_DYN0%CTY%EVEL, YDVARS%CVGQ%T0, ZRDG_MU0, ZRDG_MU0LU, ZRDG_MU0M,      &
+& ZRDG_MU0N, ZRDG_CVGQ, YDMF_PHYS_SURF%GSD_VF%PZ0F)
+
+!$ACDC }
+
+IF (LHOOK) CALL DR_HOOK('DISPATCH_ROUTINE',1,ZHOOK_HANDLE)
+
+END SUBROUTINE DISPATCH_ROUTINE

transformations/tests/test_parallel_routine_dispatch.py

@@ -0,0 +1,31 @@
+# (C) Copyright 2018- ECMWF.
+# This software is licensed under the terms of the Apache Licence Version 2.0
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+# In applying this licence, ECMWF does not waive the privileges and immunities
+# granted to it by virtue of its status as an intergovernmental organisation
+# nor does it submit to any jurisdiction.
+
+
+from pathlib import Path
+
+import pytest
+
+from loki.frontend import available_frontends, OMNI
+from loki import Sourcefile
+
+from transformations.parallel_routine_dispatch import ParallelRoutineDispatchTransformation
+
+
+@pytest.fixture(scope='module', name='here')
+def fixture_here():
+    return Path(__file__).parent
+
+
+@pytest.mark.parametrize('frontend', available_frontends(skip=[OMNI]))
+def test_parallel_routine_dispatch_parallel_regions(here, frontend):
+
+    source = Sourcefile.from_file(here/'sources/projParallelRoutineDispatch/dispatch_routine.F90', frontend=frontend)
+    transformation = ParallelRoutineDispatchTransformation()
+    transformation.apply(source['dispatch_routine'])
+
+    assert transformation.dummy_return_value == ['dispatch_routine']

transformations/transformations/parallel_routine_dispatch.py

@@ -0,0 +1,19 @@
+# (C) Copyright 2018- ECMWF.
+# This software is licensed under the terms of the Apache Licence Version 2.0
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+# In applying this licence, ECMWF does not waive the privileges and immunities
+# granted to it by virtue of its status as an intergovernmental organisation
+# nor does it submit to any jurisdiction.
+
+from loki.transform import Transformation
+
+__all__ = ['ParallelRoutineDispatchTransformation']
+
+
+class ParallelRoutineDispatchTransformation(Transformation):
+
+    def __init__(self):
+        self.dummy_return_value = []
+
+    def transform_subroutine(self, routine, **kwargs):
+        self.dummy_return_value += [routine.name.lower()]