ECWAM: add mpi launch configuration

ecmwf-ifs · Jul 26, 2024 · 2ca0706 · 2ca0706
1 parent 8945845
commit 2ca0706
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/regression_tests.yml b/.github/workflows/regression_tests.yml
@@ -40,7 +40,7 @@ jobs:
         with:
           repository: ecmwf-ifs/ecwam
           path: ecwam
-          ref: develop
+          ref: naan-loki-regr-test
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
@@ -87,4 +87,4 @@ jobs:
           OMP_STACKSIZE: 4G
         run: |
           source loki-activate
-          pytest --pyargs loki.transformations -k 'cloudsc or ecwam'
+          pytest --pyargs loki.transformations -k 'ecwam' -s
diff --git a/loki/transformations/tests/test_ecwam.py b/loki/transformations/tests/test_ecwam.py
@@ -47,7 +47,7 @@ def fixture_bundle_create(here, local_loki_bundle):
 
 @pytest.mark.usefixtures('bundle_create')
 @pytest.mark.skipif(not HAVE_FP, reason="FP needed for ECWAM parsing")
-@pytest.mark.parametrize('mode', ['idem', 'idem-stack', 'scc', 'scc-stack'])
+@pytest.mark.parametrize('mode', ['idem', 'idem-stack', 'scc', 'scc-stack', 'scc-hoist'])
 def test_ecwam(here, mode, tmp_path):
     build_dir = tmp_path/'build'
     build_cmd = [
@@ -68,7 +68,7 @@ def test_ecwam(here, mode, tmp_path):
     # Raise stack limit
     resource.setrlimit(resource.RLIMIT_STACK, (resource.RLIM_INFINITY, resource.RLIM_INFINITY))
     env = os.environ.copy()
-    env.update({'OMP_STACKSIZE': '2G', 'NVCOMPILER_ACC_CUDA_HEAPSIZE': '2G'})
+    env.update({'OMP_STACKSIZE': '2G', 'NVCOMPILER_ACC_CUDA_HEAPSIZE': '2G', 'DEV_ALLOC_SIZE': '2147483648'})
 
     # create rundir
     rundir = build_dir/'wamrun_48'