Skip to content

Add unit tests on CPU for TritonBench features #2889

Add unit tests on CPU for TritonBench features

Add unit tests on CPU for TritonBench features #2889

Workflow file for this run

name: TorchBench PR Test
on:
pull_request:
workflow_dispatch:
push:
branches:
- main
jobs:
pr-test:
env:
BASE_CONDA_ENV: "torchbench"
CONDA_ENV: "pr-ci-a100"
SETUP_SCRIPT: "/workspace/setup_instance.sh"
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
runs-on: [a100-runner]
timeout-minutes: 1440 # 24 hours
environment: docker-s3-upload
steps:
- name: Checkout TorchBench
uses: actions/checkout@v3
- name: Tune Nvidia GPU
run: |
sudo nvidia-smi -pm 1
sudo nvidia-smi -ac 1215,1410
sudo ldconfig
nvidia-smi
- name: Clone and setup Conda env
run: |
CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}"
conda create --name "${CONDA_ENV}" --clone "${BASE_CONDA_ENV}"
- name: Install TorchBench
run: |
bash ./scripts/torchbench_install.sh
- name: Validate benchmark components
run: |
bash ./scripts/torchbench_test.sh
- name: Clean up Conda env
if: always()
run: |
. "${SETUP_SCRIPT}"
conda deactivate && conda deactivate
conda remove -n "${CONDA_ENV}" --all
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
cancel-in-progress: true