Skip to content

Commit

Permalink
Secure Optimum-Benchmark CI (#11)
Browse files Browse the repository at this point in the history
* init

* revert ryzen
  • Loading branch information
IlyasMoutawwakil authored Dec 2, 2024
1 parent 1e5ae0c commit b313b8c
Showing 1 changed file with 50 additions and 0 deletions.
50 changes: 50 additions & 0 deletions .github/workflows/optimum_benchmark_instinct_ci.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
name: Optimum-Benchmark Instinct CI

on:
workflow_call:
inputs:
machine_type:
description: "Machine type to run the tests on ('single-gpu' or 'multi-gpu')"
required: true
type: string

install_extras:
description: "Install extra dependencies (e.g. 'testing,timm,diffusers')"
required: true
type: string

pytest_keywords:
description: "Pytest keywords to run (e.g. 'api and rocm')"
required: true
type: string

secrets:
HF_TOKEN:
required: false

jobs:
build:
runs-on: [self-hosted, amd-gpu, "${{ inputs.machine_type }}"]

container:
image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
options: --ipc host
--group-add video
--device /dev/kfd
--device /dev/dri
--env ROCR_VISIBLE_DEVICES

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Install dependencies
run: |
pip install -e .[${{ inputs.install_extras }}]
- name: Run tests
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
PUSH_REPO_ID: optimum-benchmark/rocm
run: |
pytest tests/* -x -s -k "${{ inputs.pytest_keywords }}"

0 comments on commit b313b8c

Please sign in to comment.