Skip to content

Commit

Permalink
Update gpu ci2
Browse files Browse the repository at this point in the history
  • Loading branch information
PaulZhang12 committed Sep 30, 2024
1 parent 9f13363 commit 124099c
Showing 1 changed file with 34 additions and 35 deletions.
69 changes: 34 additions & 35 deletions .github/workflows/unittest_ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,42 +12,41 @@ on:
jobs:
build_test:
strategy:
fail-fast: false
matrix:
os: [linux.g5.12xlarge.nvidia.gpu]
matrix:
include:
- os: linux.g5.12xlarge.nvidia.gpu
python-version: 3.9
python-tag: "py39"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: 3.9
python-tag: "py39"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.10'
python-tag: "py310"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.10'
python-tag: "py310"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.11'
python-tag: "py311"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.11'
python-tag: "py311"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.12'
python-tag: "py312"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.12'
python-tag: "py312"
cuda-tag: "cu121"
include:
- os: linux.g5.12xlarge.nvidia.gpu
python-version: 3.9
python-tag: "py39"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: 3.9
python-tag: "py39"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.10'
python-tag: "py310"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.10'
python-tag: "py310"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.11'
python-tag: "py311"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.11'
python-tag: "py311"
cuda-tag: "cu121"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.12'
python-tag: "py312"
cuda-tag: "cu118"
- os: linux.g5.12xlarge.nvidia.gpu
python-version: '3.12'
python-tag: "py312"
cuda-tag: "cu121"
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: ${{ matrix.os }}
Expand Down

0 comments on commit 124099c

Please sign in to comment.