Update test-mlperf-inference-gptj.yml

mlcommons · Sep 21, 2024 · 8c0d062 · 8c0d062
1 parent 65569d3
commit 8c0d062
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/.github/workflows/test-mlperf-inference-gptj.yml b/.github/workflows/test-mlperf-inference-gptj.yml
@@ -22,15 +22,12 @@ jobs:
  precision: [ "bfloat16" ]
 
  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: |
- python3 -m pip install cmind
+ source gh_action/bin/deactivate || python3 -m venv gh_action
+ source gh_action/bin/activate
+ export CM_REPO=$HOME/GH_CM
  cm pull repo --url=${{ github.event.pull_request.head.repo.html_url }} --checkout=${{ github.event.pull_request.head.ref }}
  - name: Test MLPerf Inference GPTJ
  run: |
- cm run script --tags=run-mlperf,inference,_submission,_short --submitter="MLCommons" --model=gptj --backend=${{ matrix.backend }} --device=cuda --scenario=Offline --test_query_count=1 --precision=${{ matrix.precision }} --target_qps=1 --quiet
+ cm run script --tags=run-mlperf,inference,_submission,_short --submitter="MLCommons" --docker --model=gptj --backend=${{ matrix.backend }} --device=cuda --scenario=Offline --test_query_count=1 --precision=${{ matrix.precision }} --target_qps=1 --quiet