mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-24 16:32:12 +00:00
9e2fdf57c0
* Removing IPEX_AVAIL. Chose to unify CPU and XPU under `ipex`. Most code is exactly similar except for a very few spots. The biggest number of spots is the kv-cache layout and the flash_xxx.py files. Since those files should be removed soon and factored away, we should not need them. * Forgot a few places. * Unrelated change. * Fixing HF_TOKEN. * HF_TOKEN
42 lines
1.2 KiB
YAML
42 lines
1.2 KiB
YAML
name: Integration tests
|
|
|
|
on:
|
|
workflow_call:
|
|
inputs:
|
|
docker_image:
|
|
type: string
|
|
description: Hardware
|
|
required: true
|
|
docker_devices:
|
|
type: string
|
|
description: Hardware
|
|
runs_on:
|
|
type: string
|
|
required: true
|
|
description: Hardware to run integration tests
|
|
jobs:
|
|
integration_tests:
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }}
|
|
cancel-in-progress: true
|
|
runs-on: ${{ inputs.runs_on }}
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@v4
|
|
- name: Inject slug/short variables
|
|
uses: rlespinasse/github-slug-action@v4.4.1
|
|
- name: Set up Python
|
|
uses: actions/setup-python@v4
|
|
with:
|
|
python-version: 3.9
|
|
- name: Install
|
|
run: |
|
|
make install-integration-tests
|
|
- name: Run tests
|
|
run: |
|
|
export DOCKER_VOLUME=/mnt/cache
|
|
export DOCKER_IMAGE=${{ inputs.docker_image }}
|
|
export DOCKER_DEVICES=${{ inputs.docker_devices }}
|
|
export HF_TOKEN=${{ secrets.HF_TOKEN }}
|
|
pytest -s -vv integration-tests
|