Skip to content

Commit

Permalink
Merge pull request #40 from mlcommons/dev
Browse files Browse the repository at this point in the history
Sync Dev
  • Loading branch information
arjunsuresh authored Jan 25, 2025
2 parents 19df660 + dd1f4c0 commit 1564ce4
Show file tree
Hide file tree
Showing 7 changed files with 924 additions and 296 deletions.
48 changes: 48 additions & 0 deletions .github/workflows/mlperf-inference-bert.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
name: MLPerf inference bert (deepsparse, tf, onnxruntime, pytorch)

on:
pull_request:
branches: [ "main", "dev" ]
paths:
- '.github/workflows/test-mlperf-inference-bert-deepsparse-tf-onnxruntime-pytorch.yml'
- '**'
- '!**.md'

jobs:
build:
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false
matrix:
# 3.12 didn't work on 20240305 - need to check
python-version: [ "3.11" ]
backend: [ "deepsparse", "tf", "onnxruntime", "pytorch" ]
precision: [ "int8", "fp32" ]
os: [ubuntu-latest, windows-latest, macos-latest]
exclude:
- backend: tf
- backend: pytorch
- backend: onnxruntime
- precision: fp32
- os: windows-latest

steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}
- name: Install mlcflow
run: |
python -m pip install --upgrade pip
python -m pip install --ignore-installed --verbose pip setuptools
python -m pip install .
mlc pull repo mlcommons@mlperf-automations --branch=dev
- name: Test MLPerf Inference Bert ${{ matrix.backend }} on ${{ matrix.os }}
if: matrix.os == 'windows-latest'
run: |
mlcr --tags=run,mlperf,inference,generate-run-cmds,_submission,_short --submitter="MLCommons" --hw_name=gh_${{ matrix.os }} --model=bert-99 --backend=${{ matrix.backend }} --device=cpu --scenario=Offline --test_query_count=5 --adr.loadgen.tags=_from-pip --pip_loadgen=yes --precision=${{ matrix.precision }} --target_qps=1 -v --quiet
- name: Test MLPerf Inference Bert ${{ matrix.backend }} on ${{ matrix.os }}
if: matrix.os != 'windows-latest'
run: |
mlcr --tags=run,mlperf,inference,generate-run-cmds,_submission,_short --submitter="MLCommons" --hw_name=gh_${{ matrix.os }}_x86 --model=bert-99 --backend=${{ matrix.backend }} --device=cpu --scenario=Offline --test_query_count=5 --precision=${{ matrix.precision }} --target_qps=1 -v --quiet
51 changes: 51 additions & 0 deletions .github/workflows/mlperf-inference-resnet50.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
name: 'MLPerf inference resnet50'

on:
pull_request:
branches: [ "main", "dev" ]
paths:
- '.github/workflows/mlperf-inference-resnet50.yml'
- '**'
- '!**.md'

jobs:
build:

strategy:
fail-fast: false
matrix:
python-version: ["3.12", "3.11", "3.8"]
on: [ubuntu-latest, macos-latest, windows-latest]
backend: [ "onnxruntime", "tf" ]
implementation: [ "python", "cpp" ]
exclude:
- backend: tf
implementation: cpp
- on: windows-latest
implementation: cpp
runs-on: "${{ matrix.on }}"
steps:
- uses: actions/checkout@v4
with:
ref: ${{ github.event.pull_request.head.sha }}
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}

- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install --ignore-installed --verbose pip setuptools
python -m pip install .
mlc pull repo mlcommons@mlperf-automations --branch=dev
- name: Test MLPerf inference ResNet50 on Windows (prebuilt loadgen)
if: runner.os == 'Windows'
run: |
mlc run script --tags=run-mlperf,inference,_submission,_short --submitter="MLCommons" --hw_name=gh_action --model=resnet50 --implementation=${{ matrix.implementation }} --backend=${{ matrix.backend }} --device=cpu --scenario=Offline --test_query_count=100 --target_qps=1 -v --quiet --adr.loadgen.tags=_from-pip --pip_loadgen=yes
- name: Test MLPerf inference ResNet50 on Unix systems
if: runner.os != 'Windows'
run: |
mlc run script --tags=run-mlperf,inference,_submission,_short --submitter="MLCommons" --hw_name=gh_action --model=resnet50 --implementation=${{ matrix.implementation }} --backend=${{ matrix.backend }} --device=cpu --scenario=Offline --test_query_count=100 --target_qps=1 -v --quiet
4 changes: 0 additions & 4 deletions config.yaml

This file was deleted.

5 changes: 5 additions & 0 deletions mlc/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
__version__ = "0.1.0"

from .main import access

__all__ = ['access']
Loading

0 comments on commit 1564ce4

Please sign in to comment.