Add Generic GPU CI.

Author: duburcqa

.github/workflows/generic.yml

@@ -10,13 +10,21 @@ concurrency:
   cancel-in-progress: ${{ github.event_name == 'pull_request' }}
 
 jobs:
-  generic-cpu:
+  generic:
     strategy:
       fail-fast: false
       matrix:
         # See official Github documentation for details: https://shorturl.at/NJgsj
-        OS: ['ubuntu-22.04', 'ubuntu-24.04', 'macos-15', 'windows-2025']
+        OS: ["ubuntu-22.04", "ubuntu-24.04", "macos-15", "windows-2025"]
         PYTHON_VERSION: ["3.10", "3.11", "3.12"]
+        PYTEST_FLAGS: ["-m required"]
+        include:
+          - OS: "macos-15-xlarge"     # (MacOS Arm - Apple M1) 6CPU+8GPU, 14GB RAM, 14GB SSD
+            PYTHON_VERSION: "3.12"
+          - OS: "gpu-t4-4-core"       # (Linux x86) 4CPU, NVIDIA Tesla T4 16 GB VRAM, 28 GB RAM, 176 GB SSD
+            PYTHON_VERSION: "3.12"
+          - OS: "win-gpu-t4-4-core"   # (Windows x86) 4CPU, NVIDIA Tesla T4 16 GB VRAM, 28 GB RAM, 176 GB SSD
+            PYTHON_VERSION: "3.12"
 
     env:
       HF_HUB_DOWNLOAD_TIMEOUT: 60
@@ -63,24 +71,30 @@ jobs:
 
           echo "PYOPENGL_PLATFORM=osmesa" >> $GITHUB_ENV
 
-      - name: Install system dependencies (Linux)
-        if: startsWith(matrix.OS, 'ubuntu-')
+      - name: Install system dependencies (Linux CPU+GPU)
+        if: startsWith(matrix.OS, 'ubuntu-') || contains(matrix.OS, 'gpu-t4-4-core')
         run: |
           sudo apt-get update
           sudo apt install -y \
               libgl1 \
               libegl1 \
-              libglvnd-dev \
-              libglew-dev \
+              libgles2 \
               libegl-dev \
-              libx11-6  \
-              libxrender1 \
               libglib2.0-0 \
               libosmesa6-dev \
-              libglx-mesa0 \
               libglu1-mesa \
+              libgl1-mesa-glx \
               libegl1-mesa-dev \
               libgles2-mesa-dev
+      - name: Install system dependencies (Linux CPU only)
+        if: startsWith(matrix.OS, 'ubuntu-')
+        run: |
+          sudo apt install -y \
+              libglew-dev \
+              libglvnd-dev \
+              libxrender1 \
+              libx11-6 \
+              libglx-mesa0
       - name: Install python dependencies
         run: |
           pip install --upgrade pip setuptools pkg-info wheel
@@ -98,7 +112,7 @@ jobs:
 
       - name: Run unit tests
         run: |
-          pytest -v --forked -m required ./tests
+          pytest -v --forked ${{ matrix.PYTEST_FLAGS }} ./tests
 
       - name: Save Updated Taichi Kernel Cache
         if: always()

.github/workflows/production.yml

@@ -14,7 +14,7 @@ concurrency:
   cancel-in-progress: ${{ github.event_name == 'pull_request' }}
 
 jobs:
-  linux-gpu:
+  production:
     runs-on: [self-hosted, coreweave]
 
     env: