tweak

Author: ggorman

.github/workflows/pytest-gpu.yml

@@ -38,12 +38,7 @@ jobs:
       fail-fast: false
       matrix:
         name: [pytest-gpu-acc-nvidia, pytest-gpu-omp-amd]
-        test_examples:
-          - examples/seismic/tti/tti_example.py
-          - examples/seismic/acoustic/acoustic_example.py
-          - examples/seismic/viscoacoustic/viscoacoustic_example.py
-          - examples/seismic/viscoelastic/viscoelastic_example.py
-          - examples/seismic/elastic/elastic_example.py
+        test_examples: ["examples/seismic/tti/tti_example.py examples/seismic/acoustic/acoustic_example.py examples/seismic/viscoacoustic/viscoacoustic_example.py examples/seismic/viscoelastic/viscoelastic_example.py examples/seismic/elastic/elastic_example.py"]
 
         include:
           # -------------------- NVIDIA job --------------------
@@ -159,17 +154,16 @@ jobs:
           docker run ${{ matrix.flags }} --env DEVITO_MPI=1 "${DOCKER_IMAGE}" \
             mpiexec -n 2 pytest ${{ matrix.test_examples }}
 
-      - name: Clean up test image
+      - name: Builder & image cleanup (keep 3 days of cache)
         if: always()
         run: |
-          # Remove only the image we just built
+          # Remove only the test image we built
           docker rmi -f "${DOCKER_IMAGE}" || true
 
-          # Guard global prune the layers we created
-          # (label ci-run=$GITHUB_RUN_ID)
+          # Classic image layers created in this job
           docker image prune -f --filter label=ci-run=$GITHUB_RUN_ID
 
-          docker builder prune -f --filter "name=$RUNNER_NAME" --filter "until=168h"
-
-          # May be overkill - commented out for now but left here as a reminder
-          # docker buildx rm $RUNNER_NAME || true
+          # BuildKit cache: target the per-runner builder explicitly
+          docker builder prune --builder "$RUNNER_NAME" \
+                              -f \
+                              --filter "until=72h"