Skip to content

Commit 1c49242

Browse files
committed
[DROP BEFORE MERGE] Limit firedrake-check to a single GPU test and run nprocs=1 Firedrake tests on GPU CI
1 parent e54b655 commit 1c49242

2 files changed

Lines changed: 13 additions & 3 deletions

File tree

.github/workflows/core.yml

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ jobs:
491491
PYOP2_CFLAGS: -fno-fast-math
492492
# NOTE: One should occasionally update test_durations.json by running
493493
# 'make test_durations' inside a 'firedrake:latest' Docker image.
494-
EXTRA_PYTEST_ARGS: --splitting-algorithm least_duration --timeout=600 --timeout-method=thread -o faulthandler_timeout=660 --durations-path=./firedrake-repo/tests/test_durations.json --durations=50
494+
EXTRA_PYTEST_ARGS: -m skipslepc --splitting-algorithm least_duration --timeout=600 --timeout-method=thread -o faulthandler_timeout=660 --durations-path=./firedrake-repo/tests/test_durations.json --durations=50
495495
PYTEST_MPI_MAX_NPROCS: 8
496496
# Prevent PETSc from exiting with an error due to using non-GPU aware system MPI
497497
PETSC_OPTIONS: -use_gpu_aware_mpi 0
@@ -588,7 +588,8 @@ jobs:
588588
589589
pip install --verbose $EXTRA_PIP_FLAGS \
590590
--no-binary h5py \
591-
'./firedrake-repo[check]'
591+
--extra-index-url https://download.pytorch.org/whl/cu130 \
592+
'./firedrake-repo[ci]'
592593
593594
firedrake-clean
594595
pip list
@@ -622,6 +623,15 @@ jobs:
622623
echo "SNESSolve not found in PETSc log_view output!"
623624
exit 1
624625
626+
- name: Run Firedrake tests (nprocs = 1)
627+
if: inputs.run_tests && (success() || steps.install.conclusion == 'success')
628+
run: |
629+
. venv/bin/activate
630+
: # Use pytest-xdist here so we can have a single collated output (not possible
631+
: # for parallel tests)
632+
firedrake-run-split-tests 1 1 -n 8 "$EXTRA_PYTEST_ARGS" firedrake-repo/tests/firedrake
633+
timeout-minutes: 90
634+
625635
- name: Post-run cleanup
626636
if: always()
627637
run: |

scripts/firedrake-check

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ TESTS = {
2424
# near nullspace
2525
"tests/firedrake/regression/test_nullspace.py::test_near_nullspace",
2626
# GPU offload
27-
"tests/firedrake/offload/test_poisson_offloading_pc.py::test_poisson_offload",
27+
"tests/firedrake/offload/test_poisson_offloading_pc.py::test_poisson_offload[cg-gamg-True]",
2828
),
2929
2: (
3030
# HDF5/checkpointing

0 commit comments

Comments
 (0)