From fa7a78a6728d924f8e90c27e8faf6e2451c7b740 Mon Sep 17 00:00:00 2001 From: Milo Cress Date: Wed, 1 May 2024 18:10:59 -0400 Subject: [PATCH] Uncomment GPU tests (#1162) * Uncomment GPU tests * Update pr-gpu.yaml --- .github/workflows/pr-gpu.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index ebcb522f67..2c4aeb3d4c 100644 --- a/.github/workflows/pr-gpu.yaml +++ b/.github/workflows/pr-gpu.yaml @@ -9,7 +9,6 @@ on: - main - release/** workflow_dispatch: -# Cancel old runs when a new commit is pushed to the same branch if not on main or dev concurrency: group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }} cancel-in-progress: ${{ github.ref != 'refs/heads/main' }} @@ -17,6 +16,7 @@ jobs: pytest-gpu: uses: mosaicml/ci-testing/.github/workflows/pytest-gpu.yaml@v0.0.5 strategy: + fail-fast: false matrix: include: - name: "gpu-2.2.1" @@ -24,11 +24,11 @@ jobs: markers: "gpu" pip_deps: "[all]" pytest_command: "coverage run -m pytest" - # - name: "gpu-2.3.0" - # container: mosaicml/pytorch:2.3.0_cu121-python3.11-ubuntu20.04 - # markers: "gpu" - # pytest_command: "coverage run -m pytest" - # deps_group: "all" + - name: "gpu-2.3.0" + container: mosaicml/pytorch:2.3.0_cu121-python3.11-ubuntu20.04 + markers: "gpu" + pytest_command: "coverage run -m pytest" + deps_group: "all" name: ${{ matrix.name }} if: github.repository_owner == 'mosaicml' with: