#2971 update test imports and __init__.py #5269
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # ----------------------------------------------------------------------------- | |
| # BSD 3-Clause License | |
| # | |
| # Copyright (c) 2023-2025, Science and Technology Facilities Council. | |
| # All rights reserved. | |
| # | |
| # Redistribution and use in source and binary forms, with or without | |
| # modification, are permitted provided that the following conditions are met: | |
| # | |
| # * Redistributions of source code must retain the above copyright notice, this | |
| # list of conditions and the following disclaimer. | |
| # | |
| # * Redistributions in binary form must reproduce the above copyright notice, | |
| # this list of conditions and the following disclaimer in the documentation | |
| # and/or other materials provided with the distribution. | |
| # | |
| # * Neither the name of the copyright holder nor the names of its | |
| # contributors may be used to endorse or promote products derived from | |
| # this software without specific prior written permission. | |
| # | |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS | |
| # FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE | |
| # COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, | |
| # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, | |
| # BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; | |
| # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | |
| # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
| # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN | |
| # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | |
| # POSSIBILITY OF SUCH DAMAGE. | |
| # ----------------------------------------------------------------------------- | |
| # Author S. Siso, STFC Daresbury Lab | |
| # This workflow will use a self-hosted runner to perform the more expensive | |
| # integrations tests that are not run on GHA systems. | |
| name: LFRic Integration Tests | |
| on: | |
| push | |
| jobs: | |
| run_if_on_mirror: | |
| if: ${{ github.repository == 'stfc/PSyclone-mirror' }} | |
| runs-on: self-hosted | |
| env: | |
| LFRIC_APPS_REV: 10798 | |
| LFRIC_CORE_REV: 52772 | |
| PYTHON_VERSION: 3.13.3 | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| submodules: recursive | |
| # This is required to get the commit history for merge commits for | |
| # the ci-skip check below. | |
| fetch-depth: '0' | |
| - name: Check for [skip ci] in commit message | |
| uses: mstachniuk/ci-skip@v1 | |
| with: | |
| # This setting causes the tests to 'fail' if [skip ci] is specified | |
| fail-fast: true | |
| commit-filter: '[skip ci]' | |
| - name: Install dependencies | |
| run: | | |
| # Use a specific version of Python (rather than the system-wide one). | |
| module load python/${PYTHON_VERSION} | |
| python -m venv .runner_venv | |
| . .runner_venv/bin/activate | |
| python -m pip install --upgrade pip | |
| # Uncomment the below to use the submodule version of fparser rather | |
| # than the latest release from pypi. | |
| # pip install external/fparser | |
| pip install .[test] | |
| # Fetch the specified version of LFRic apps | |
| source /apps/spack/psyclone-spack-25.3/spack-repo/share/spack/setup-env.sh | |
| spack load fcm%gcc | |
| fcm co -r ${LFRIC_APPS_REV} https://code.metoffice.gov.uk/svn/lfric_apps/main/trunk/ lfric_apps | |
| # We need to patch lfric_core to resolve issues with the nvfortran compiler | |
| # see https://code.metoffice.gov.uk/trac/lfric/ticket/4621 | |
| fcm co -r ${LFRIC_CORE_REV} https://code.metoffice.gov.uk/svn/lfric/LFRic/trunk/ lfric_core | |
| sed -i "s|lfric_core_rev=$LFRIC_CORE_REV|lfric_core_rev=|g" lfric_apps/dependencies.sh | |
| sed -i "s|lfric_core_sources=|lfric_core_sources=$PWD/lfric_core|g" lfric_apps/dependencies.sh | |
| cd lfric_core | |
| svn patch ${GITHUB_WORKSPACE}/examples/lfric/scripts/KGOs/nvidia_lfric_core.patch | |
| # PSyclone, compile and run MetOffice LFRic with 6 MPI ranks. | |
| # COMPUTE_ANNEXED_DOFS is set to false for this test by using the | |
| # appropriate cfg file. | |
| - name: LFRic passthrough (with DistributedMemory) | |
| run: | | |
| # Set up environment | |
| source /apps/spack/psyclone-spack-25.3/spack-repo/share/spack/setup-env.sh | |
| spack load lfric-build-environment%gcc | |
| spack load gcc | |
| source .runner_venv/bin/activate | |
| export PSYCLONE_LFRIC_DIR=${GITHUB_WORKSPACE}/examples/lfric/scripts | |
| export PSYCLONE_CONFIG_FILE=${PSYCLONE_LFRIC_DIR}/KGOs/lfric_psyclone_no_annexed.cfg | |
| export LFRIC_DIR=${GITHUB_WORKSPACE}/lfric_apps | |
| cd ${LFRIC_DIR} | |
| # Ensure no transformations are applied by supplying an empty | |
| # transformation script. | |
| export OPT_DIR=${LFRIC_DIR}/applications/gungho_model/optimisation/psyclone-test/psykal | |
| mkdir -p ${OPT_DIR} | |
| cp ${PSYCLONE_LFRIC_DIR}/does_nothing.py ${OPT_DIR}/global.py | |
| # Clean previous version and compile again | |
| rm -rf applications/gungho_model/working | |
| ./build/local_build.py -a gungho_model -v -p psyclone-test | |
| # Run | |
| cd applications/gungho_model/example | |
| rm -f timer.txt gungho_model-checksums.txt # In case there were from a previous run | |
| cp ${PSYCLONE_LFRIC_DIR}/KGOs/configuration_c16_10it.nml . | |
| # Number of OMP threads have to be set even when psyclone does not inject OpenMP | |
| # because some psykal-lite routines have it. | |
| export OMP_NUM_THREADS=1 | |
| mpirun -n 6 ../bin/gungho_model configuration_c16_10it.nml | |
| python ${PSYCLONE_LFRIC_DIR}/compare_ouput.py ${PSYCLONE_LFRIC_DIR}/KGOs/gungho_model_c16_10it-checksums.txt gungho_model-checksums.txt | |
| cat timer.txt | |
| export VAR_TIME=$(grep "gungho_model" timer.txt | cut -d'|' -f5) | |
| ${HOME}/mongosh-2.1.1-linux-x64/bin/mongosh \ | |
| "mongodb+srv://cluster0.x8ncpxi.mongodb.net/PerformanceMonitoring" \ | |
| --quiet --apiVersion 1 --username ${{ secrets.MONGODB_USERNAME }} \ | |
| --password ${{ secrets.MONGODB_PASSWORD }} \ | |
| --eval 'db.GitHub_CI.insertOne({branch_name: "'"$GITHUB_REF_NAME"'", commit: "'"$GITHUB_SHA"'", | |
| github_job: "'"$GITHUB_RUN_ID"'"-"'"$GITHUB_RUN_ATTEMPT"'", | |
| ci_test: "LFRic Passthrough with DM", lfric_apps_version: '"$LFRIC_APPS_REV"', system: "GlaDos", | |
| compiler:"spack-gfortran-14", date: new Date(), elapsed_time: '"$VAR_TIME"'})' | |
| # PSyclone, compile and run MetOffice gungho_model on GPU | |
| - name: LFRic GungHo with OpenMP offload | |
| run: | | |
| # Set up environment | |
| source /apps/spack/psyclone-spack-25.3/spack-repo/share/spack/setup-env.sh | |
| spack load lfric-build-environment%nvhpc | |
| spack load nvhpc | |
| source .runner_venv/bin/activate | |
| export PSYCLONE_LFRIC_DIR=${GITHUB_WORKSPACE}/examples/lfric/scripts | |
| export PSYCLONE_CONFIG_FILE=${PSYCLONE_LFRIC_DIR}/KGOs/lfric_psyclone.cfg | |
| export LFRIC_DIR=${GITHUB_WORKSPACE}/lfric_apps | |
| cd ${LFRIC_DIR} | |
| # PSyclone scripts must now be under 'optimisation/psykal' and be called 'global.py' | |
| export OPT_DIR=${LFRIC_DIR}/applications/gungho_model/optimisation/psyclone-test/psykal | |
| mkdir -p ${OPT_DIR} | |
| cp ${PSYCLONE_LFRIC_DIR}/gpu_offloading.py ${OPT_DIR}/global.py | |
| # Clean previous version and compile again | |
| rm -rf applications/gungho_model/working | |
| LFRIC_OFFLOAD_DIRECTIVES=omp ./build/local_build.py -a gungho_model -p psyclone-test | |
| cd applications/gungho_model/example | |
| rm -f timer.txt gungho_model-checksums.txt # In case there were from a previous run | |
| cp ${PSYCLONE_LFRIC_DIR}/KGOs/configuration_c16_10it.nml . | |
| export OMP_NUM_THREADS=1 | |
| mpirun -n 1 ../bin/gungho_model configuration_c16_10it.nml | |
| python ${PSYCLONE_LFRIC_DIR}/compare_ouput.py ${PSYCLONE_LFRIC_DIR}/KGOs/gungho_model_c16_10it-checksums.txt gungho_model-checksums.txt | |
| cat timer.txt | |
| export VAR_TIME=$(grep "gungho_model" timer.txt | cut -d'|' -f5) | |
| ${HOME}/mongosh-2.1.1-linux-x64/bin/mongosh \ | |
| "mongodb+srv://cluster0.x8ncpxi.mongodb.net/PerformanceMonitoring" \ | |
| --quiet --apiVersion 1 --username ${{ secrets.MONGODB_USERNAME }} \ | |
| --password ${{ secrets.MONGODB_PASSWORD }} \ | |
| --eval 'db.GitHub_CI.insertOne({branch_name: "'"$GITHUB_REF_NAME"'", commit: "'"$GITHUB_SHA"'", | |
| github_job: "'"$GITHUB_RUN_ID"'"-"'"$GITHUB_RUN_ATTEMPT"'", | |
| ci_test: "LFRic OpenMP offloading", lfric_apps_version: '"$LFRIC_APPS_REV"', system: "GlaDos", | |
| compiler:"spack-nvhpc-24.11", date: new Date(), elapsed_time: '"$VAR_TIME"'})' | |
| # PSyclone, compile and run MetOffice gungho_model on GPU | |
| - name: LFRic GungHo with OpenACC offload | |
| run: | | |
| # Set up environment | |
| source /apps/spack/psyclone-spack-25.3/spack-repo/share/spack/setup-env.sh | |
| spack load lfric-build-environment%nvhpc | |
| spack load nvhpc | |
| source .runner_venv/bin/activate | |
| export PSYCLONE_LFRIC_DIR=${GITHUB_WORKSPACE}/examples/lfric/scripts | |
| export PSYCLONE_CONFIG_FILE=${PSYCLONE_LFRIC_DIR}/KGOs/lfric_psyclone.cfg | |
| export LFRIC_DIR=${GITHUB_WORKSPACE}/lfric_apps | |
| cd ${LFRIC_DIR} | |
| # PSyclone scripts must now be under 'optimisation/psykal' and be called 'global.py' | |
| export OPT_DIR=${LFRIC_DIR}/applications/gungho_model/optimisation/psyclone-test/psykal | |
| mkdir -p ${OPT_DIR} | |
| cp ${PSYCLONE_LFRIC_DIR}/gpu_offloading.py ${OPT_DIR}/global.py | |
| # Clean previous version and compile again | |
| rm -rf applications/gungho_model/working | |
| LFRIC_OFFLOAD_DIRECTIVES=acc ./build/local_build.py -a gungho_model -p psyclone-test | |
| cd applications/gungho_model/example | |
| rm -f timer.txt gungho_model-checksums.txt # In case there were from a previous run | |
| cp ${PSYCLONE_LFRIC_DIR}/KGOs/configuration_c16_10it.nml . | |
| export OMP_NUM_THREADS=1 | |
| mpirun -n 1 ../bin/gungho_model configuration_c16_10it.nml | |
| python ${PSYCLONE_LFRIC_DIR}/compare_ouput.py ${PSYCLONE_LFRIC_DIR}/KGOs/gungho_model_c16_10it-checksums.txt gungho_model-checksums.txt | |
| cat timer.txt | |
| export VAR_TIME=$(grep "gungho_model" timer.txt | cut -d'|' -f5) | |
| ${HOME}/mongosh-2.1.1-linux-x64/bin/mongosh \ | |
| "mongodb+srv://cluster0.x8ncpxi.mongodb.net/PerformanceMonitoring" \ | |
| --quiet --apiVersion 1 --username ${{ secrets.MONGODB_USERNAME }} \ | |
| --password ${{ secrets.MONGODB_PASSWORD }} \ | |
| --eval 'db.GitHub_CI.insertOne({branch_name: "'"$GITHUB_REF_NAME"'", commit: "'"$GITHUB_SHA"'", | |
| github_job: "'"$GITHUB_RUN_ID"'"-"'"$GITHUB_RUN_ATTEMPT"'", | |
| ci_test: "LFRic OpenACC", lfric_apps_version: '"$LFRIC_APPS_REV"', system: "GlaDos", | |
| compiler:"spack-nvhpc-24.11", date: new Date(), elapsed_time: '"$VAR_TIME"'})' | |
| # PSyclone, compile and run MetOffice LFRic with all optimisations and 6 mpi ranks each with 2 OpenMP threads | |
| - name: LFRic with all transformations | |
| run: | | |
| # Set up environment | |
| source /apps/spack/psyclone-spack-25.3/spack-repo/share/spack/setup-env.sh | |
| spack load lfric-build-environment%gcc | |
| spack load gcc | |
| source .runner_venv/bin/activate | |
| export PSYCLONE_LFRIC_DIR=${GITHUB_WORKSPACE}/examples/lfric/scripts | |
| export PSYCLONE_CONFIG_FILE=${PSYCLONE_LFRIC_DIR}/KGOs/lfric_psyclone.cfg | |
| export LFRIC_DIR=${GITHUB_WORKSPACE}/lfric_apps | |
| cd ${LFRIC_DIR} | |
| # PSyclone scripts must now be under 'optimisation/psykal' and be called 'global.py' | |
| export OPT_DIR=${LFRIC_DIR}/applications/gungho_model/optimisation/psyclone-test/psykal | |
| mkdir -p ${OPT_DIR} | |
| cp ${PSYCLONE_LFRIC_DIR}/everything_everywhere_all_at_once.py ${OPT_DIR}/global.py | |
| # Clean previous version and compile again | |
| rm -rf applications/gungho_model/working | |
| ./build/local_build.py -a gungho_model -p psyclone-test -v | |
| # Run | |
| cd applications/gungho_model/example | |
| rm -f timer.txt gungho_model-checksums.txt # In case there were from a previous run | |
| cp ${PSYCLONE_LFRIC_DIR}/KGOs/configuration_c16_10it.nml . | |
| export OMP_NUM_THREADS=2 | |
| mpirun -n 6 ../bin/gungho_model configuration_c16_10it.nml | |
| python ${PSYCLONE_LFRIC_DIR}/compare_ouput.py ${PSYCLONE_LFRIC_DIR}/KGOs/gungho_model_c16_10it-checksums.txt gungho_model-checksums.txt | |
| cat timer.txt | |
| export VAR_TIME=$(grep "gungho_model" timer.txt | cut -d'|' -f5) | |
| ${HOME}/mongosh-2.1.1-linux-x64/bin/mongosh \ | |
| "mongodb+srv://cluster0.x8ncpxi.mongodb.net/PerformanceMonitoring" \ | |
| --quiet --apiVersion 1 --username ${{ secrets.MONGODB_USERNAME }} \ | |
| --password ${{ secrets.MONGODB_PASSWORD }} \ | |
| --eval 'db.GitHub_CI.insertOne({branch_name: "'"$GITHUB_REF_NAME"'", commit: "'"$GITHUB_SHA"'", | |
| github_job: "'"$GITHUB_RUN_ID"'"-"'"$GITHUB_RUN_ATTEMPT"'", | |
| ci_test: "LFRic all transformations", lfric_version: '"$LFRIC_APPS_REV"', omp_threads: '"$OMP_NUM_THREADS"', | |
| system: "GlaDos", compiler:"spack-gfortran-14", date: new Date(), elapsed_time: '"$VAR_TIME"'})' | |
| # PSyclone, compile and run MetOffice LFRic with all optimisations and 6 mpi ranks each with 2 OpenMP threads | |
| - name: LFRic with colour-tiling and OpenMP | |
| run: | | |
| # Set up environment | |
| source /apps/spack/psyclone-spack/spack-repo/share/spack/setup-env.sh | |
| spack load lfric-build-environment%gcc | |
| spack load gcc | |
| source .runner_venv/bin/activate | |
| export PSYCLONE_LFRIC_DIR=${GITHUB_WORKSPACE}/examples/lfric/scripts | |
| export PSYCLONE_CONFIG_FILE=${PSYCLONE_LFRIC_DIR}/KGOs/lfric_psyclone.cfg | |
| export LFRIC_DIR=${GITHUB_WORKSPACE}/lfric_apps | |
| cd ${LFRIC_DIR} | |
| # PSyclone scripts must now be under 'optimisation/psykal' and be called 'global.py' | |
| export OPT_DIR=${LFRIC_DIR}/applications/gungho_model/optimisation/psyclone-test/psykal | |
| mkdir -p ${OPT_DIR} | |
| cp ${PSYCLONE_LFRIC_DIR}/tiledcolouring_and_omp.py ${OPT_DIR}/global.py | |
| # Clean previous version and compile again | |
| rm -rf applications/gungho_model/working | |
| ./build/local_build.py -a gungho_model -p psyclone-test -v | |
| # Run | |
| cd applications/gungho_model/example | |
| rm -f timer.txt gungho_model-checksums.txt # In case there were from a previous run | |
| cp ${PSYCLONE_LFRIC_DIR}/KGOs/configuration_c16_10it.nml . | |
| export OMP_NUM_THREADS=2 | |
| mpirun -n 6 ../bin/gungho_model configuration_c16_10it.nml | |
| python ${PSYCLONE_LFRIC_DIR}/compare_ouput.py ${PSYCLONE_LFRIC_DIR}/KGOs/gungho_model_c16_10it-checksums.txt gungho_model-checksums.txt | |
| cat timer.txt | |
| export VAR_TIME=$(grep "gungho_model" timer.txt | cut -d'|' -f5) | |
| ${HOME}/mongosh-2.1.1-linux-x64/bin/mongosh \ | |
| "mongodb+srv://cluster0.x8ncpxi.mongodb.net/PerformanceMonitoring" \ | |
| --quiet --apiVersion 1 --username ${{ secrets.MONGODB_USERNAME }} \ | |
| --password ${{ secrets.MONGODB_PASSWORD }} \ | |
| --eval 'db.GitHub_CI.insertOne({branch_name: "'"$GITHUB_REF_NAME"'", commit: "'"$GITHUB_SHA"'", | |
| github_job: "'"$GITHUB_RUN_ID"'"-"'"$GITHUB_RUN_ATTEMPT"'", | |
| ci_test: "LFRic colour-tiling", lfric_version: '"$LFRIC_APPS_REV"', omp_threads: '"$OMP_NUM_THREADS"', | |
| system: "GlaDos", compiler:"spack-gfortran-14", date: new Date(), elapsed_time: '"$VAR_TIME"'})' |