Skip to content

Bump version (#278) #823

Bump version (#278)

Bump version (#278) #823

name: CLI CUDA OnnxRuntime Tests
on:
workflow_dispatch:
push:
branches:
- main
pull_request:
branches:
- main
types:
- opened
- reopened
- synchronize
- labeled
- unlabeled
concurrency:
cancel-in-progress: true
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
jobs:
run_cli_cuda_onnxruntime_tests:
if: ${{
(github.event_name == 'push') ||
(github.event_name == 'workflow_dispatch') ||
contains( github.event.pull_request.labels.*.name, 'cli') ||
contains( github.event.pull_request.labels.*.name, 'cuda') ||
contains( github.event.pull_request.labels.*.name, 'onnxruntime') ||
contains( github.event.pull_request.labels.*.name, 'cli_cuda_onnxruntime')
}}
runs-on: [single-gpu, nvidia-gpu, a10, ci]
container:
image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
options: --ipc host --gpus all
steps:
- name: Checkout
uses: actions/checkout@v4
- name: Install dependencies
run: |
pip install -e .[testing,onnxruntime-gpu,diffusers,timm]
- name: Run tests
run: |
pytest tests/test_cli.py -x -s -k "cli and cuda and onnxruntime"