Skip to content

Commit

Permalink
Torch Nightly Smoke Test (pytorch#183)
Browse files Browse the repository at this point in the history
* Torch Nightly Smoke test
  • Loading branch information
msaroufim authored Apr 26, 2024
1 parent 739e62d commit 8f67d29
Showing 1 changed file with 40 additions and 0 deletions.
40 changes: 40 additions & 0 deletions .github/workflows/nightly_smoke_test.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
name: PyTorch CUDA Nightly Smoke Test

on:
schedule:
# 6 am PST every day
- cron: "0 14 * * *"
workflow_dispatch:

concurrency:
group: regression_test-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
cancel-in-progress: true

env:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}

jobs:
test:
strategy:
fail-fast: false
matrix:
include:
- name: CUDA Nightly
runs-on: linux.g5.12xlarge.nvidia.gpu
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
gpu-arch-type: "cuda"
gpu-arch-version: "12.1"


uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: ${{ matrix.runs-on }}
gpu-arch-type: ${{ matrix.gpu-arch-type }}
gpu-arch-version: ${{ matrix.gpu-arch-version }}
script: |
python -m pip install --upgrade pip
pip install ${{ matrix.torch-spec }}
pip install -r requirements.txt
pip install -r dev-requirements.txt
python setup.py install
pytest test --verbose -s

0 comments on commit 8f67d29

Please sign in to comment.