workflows

.github/workflows/test.yml

@@ -10,23 +10,34 @@ on: [ push ]
 jobs:
   build:
     name: Build Test
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.runner.labels }}
+    container: ${{ matrix.runner.container }}
     strategy:
       fail-fast: false
       matrix:
-        python-version: [ "3.10", "3.11", "3.12" ]
+        runner:
+          - labels: [self-hosted, Linux, X64, cpu]
+            container: "ubuntu"
+          - labels: [self-hosted, Linux, X64, ipex-a380-6gb]
+            container: "intel/intel-optimized-pytorch:2.3.0-pip-base"
+          - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
+            container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
+          - labels: [self-hosted, Linux, X64, cuda-3060-12gb]
+            container: "nvcr.io/nvidia/pytorch:24.04-py3"
     steps:
+      - run: |
+          apt update
+          apt install -y python3 python3-setuptools git build-essential ffmpeg libsm6 libxext6
+          pip uninstall -y $(pip list --format=freeze | grep opencv) &&
+              rm -rf /usr/local/lib/python3.10/dist-packages/cv2/ || rm -rf /usr/local/lib/python3.11/dist-packages/cv2/ || rm -rf /usr/local/lib/python3.12/dist-packages/cv2/
       - uses: actions/checkout@v4
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v4
-        with:
-          python-version: ${{ matrix.python-version }}
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install .[dev]
+          pip install --no-build-isolation .[dev]
       - name: Run unit tests
         run: |
+          export HSA_OVERRIDE_GFX_VERSION=11.0.0
           pytest -v tests/unit
       - name: Lint for errors
         run: |