diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 8841877ea..0f7a1712b 100644 --- a/.github/workflows/test.yml +++ b/.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: |