diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml index 9b2334a95..67e310435 100644 --- a/.github/workflows/gpu-ci.yml +++ b/.github/workflows/gpu-ci.yml @@ -2,17 +2,24 @@ name: GPU CI on: workflow_dispatch: - pull_request: - branches: [main] - types: [opened, synchronize, reopened] - -concurrency: - group: ${{ github.workflow }}-${{ github.ref }} - cancel-in-progress: true + push: + branches: + - main + - "pull-request/[0-9]+" + tags: + - "v[0-9]+.[0-9]+.[0-9]+" jobs: gpu-ci-premerge: - runs-on: 1GPU + runs-on: linux-amd64-gpu-p100-latest-1 + container: + image: nvcr.io/nvstaging/merlin/merlin-ci-runner:latest + env: + NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }} + options: --shm-size=1G + credentials: + username: $oauthtoken + password: ${{ secrets.NGC_TOKEN }} steps: - uses: actions/checkout@v3 @@ -27,4 +34,4 @@ jobs: raw=$(git branch -r --contains ${{ github.ref_name }}) branch=${raw/origin\/} fi - cd ${{ github.workspace }}; tox -e test-gpu -- $branch + tox -e test-gpu -- $branch