Skip to content

Commit 113f86a

Browse files
committed
Add Generic GPU CI.
1 parent 4a7ec67 commit 113f86a

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

.github/workflows/generic.yml

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,16 @@ jobs:
1515
fail-fast: false
1616
matrix:
1717
# See official Github documentation for details: https://shorturl.at/NJgsj
18-
OS: ['ubuntu-22.04', 'ubuntu-24.04', 'macos-15', 'windows-2025']
18+
OS: ["ubuntu-22.04", "ubuntu-24.04", "macos-15", "windows-2025"]
1919
PYTHON_VERSION: ["3.10", "3.11", "3.12"]
20+
REQUIRED_ONLY: "1"
21+
include:
22+
- OS: "macos-15-xlarge" # (MacOS Arm - Apple M1) 6CPU+8GPU, 14GB RAM, 14GB SSD
23+
PYTHON_VERSION: "3.12"
24+
- OS: "gpu-t4-4-core" # (Linux x86) 4CPU, NVIDIA Tesla T4 16 GB VRAM, 28 GB RAM, 176 GB SSD
25+
PYTHON_VERSION: "3.12"
26+
- OS: "win-gpu-t4-4-core" # (Windows x86) 4CPU, NVIDIA Tesla T4 16 GB VRAM, 28 GB RAM, 176 GB SSD
27+
PYTHON_VERSION: "3.12"
2028

2129
env:
2230
HF_HUB_DOWNLOAD_TIMEOUT: 60
@@ -98,7 +106,11 @@ jobs:
98106
99107
- name: Run unit tests
100108
run: |
101-
pytest -v --forked -m required ./tests
109+
if [[ "${{ matrix.REQUIRED_ONLY }}" == "1" ]] ; then
110+
PYTEST_FLAGS="-m required"
111+
fi
112+
113+
pytest -v --forked ${PYTEST_FLAGS} ./tests
102114
103115
- name: Save Updated Taichi Kernel Cache
104116
if: always()

0 commit comments

Comments
 (0)