Skip to content

Add aarch64-Compatible Base Image for MLPerf Inference & Fix TensorRT Version Matching #648

Add aarch64-Compatible Base Image for MLPerf Inference & Fix TensorRT Version Matching

Add aarch64-Compatible Base Image for MLPerf Inference & Fix TensorRT Version Matching #648

# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
name: CM tutorial tvm pip install
on:
pull_request:
branches: [ "main", "test", "mlperf-inference" ]
paths:
- '.github/workflows/test-cm-tutorial-tvm-pip.yml'
- '**'
- '!**.md'
jobs:
test_vm_runtime:
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
python-version: ["3.9"]
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 cmind
cm pull repo --url=${{ github.event.pull_request.head.repo.html_url }} --checkout=${{ github.event.pull_request.head.ref }}
cm run script --quiet --tags=get,sys-utils-cm
- name: Test CM Tutorial TVM pip install with VirtualMachine Runtime
run: |
python script/test-cm-core/src/tutorials/test_tutorial_tvm_pip_vm.py
test_ge_runtime:
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
python-version: ["3.9"]
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 dependencies
run: |
python -m pip install cmind
cm pull repo --url=${{ github.event.pull_request.head.repo.html_url }} --checkout=${{ github.event.pull_request.head.ref }}
cm run script --quiet --tags=get,sys-utils-cm
- name: Test CM Tutorial TVM pip install with GraphExecutor Runtime
run: |
python script/test-cm-core/src/tutorials/test_tutorial_tvm_pip_ge.py