Skip to content

[TEST] test only

[TEST] test only #4919

Workflow file for this run

name: IPEX - Test
on:
push:
branches:
- main
- v*-release
pull_request:
branches:
- main
concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
jobs:
build:
strategy:
fail-fast: false
matrix:
torch-version: ["2.6.0"]
transformers-version: ["4.47.*"]
runs-on: ubuntu-latest
container: intel/intel-extension-for-pytorch:2.6.0-pip-base
steps:
- name: Install dependencies
run: |
pip install --upgrade pip
pip install .[ipex,tests] transformers[testing]==${{ matrix.transformers-version }}
- name: Install bitsandbytes
run: |
git clone --branch multi-backend-refactor https://github.com/bitsandbytes-foundation/bitsandbytes.git
cd bitsandbytes
pip install .
- name: Assert versions
run: |
python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
python -c "import intel_extension_for_pytorch; print(intel_extension_for_pytorch.__version__); assert intel_extension_for_pytorch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
python -c "import transformers; print(transformers.__version__); assert transformers.__version__.startswith('${{ matrix.transformers-version }}'.replace('.*', ''))"
- name: Test with Pytest
run: |
pytest tests/ipex