Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add nightly and weekly whl tests #760

Merged
merged 12 commits into from
Aug 16, 2024
17 changes: 14 additions & 3 deletions .github/actions/inductor-xpu-e2e-test/action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,11 @@ inputs:
hf_token:
required: false
description: HUGGING_FACE_HUB_TOKEN for torchbench test
test_type:
mengfei25 marked this conversation as resolved.
Show resolved Hide resolved
required: false
type: string
default: 'lts'
description: test type, lts/rolling/whl

runs:
using: composite
Expand All @@ -45,10 +50,14 @@ runs:
if [[ ${{ inputs.suite }} == *"torchbench"* ]]; then
cd ../ && rm -rf audio && git clone --single-branch -b main https://github.com/pytorch/audio.git
cd audio && git checkout $TORCHAUDIO_COMMIT_ID
python setup.py bdist_wheel && pip uninstall torchaudio -y && pip install dist/*.whl
if [ ${{ inputs.test_type }} != "whl" ]; then
python setup.py bdist_wheel && pip uninstall torchaudio -y && pip install dist/*.whl
fi
cd ../ && rm -rf vision && git clone --single-branch -b main https://github.com/pytorch/vision.git
cd vision && git checkout $TORCHVISION_COMMIT_ID
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl
if [ ${{ inputs.test_type }} != "whl" ]; then
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl
fi
cd ../ && python -c "import torch, torchvision, torchaudio"
rm -rf benchmark && git clone https://github.com/pytorch/benchmark.git
cd benchmark && git checkout $TORCHBENCH_COMMIT_ID && pip install --no-deps -r requirements.txt
Expand All @@ -65,7 +74,9 @@ runs:
if [[ ${{ inputs.suite }} == *"timm_models"* ]]; then
cd ../ && rm -rf vision && git clone --single-branch -b main https://github.com/pytorch/vision.git
cd vision && git checkout $TORCHVISION_COMMIT_ID
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl
if [ ${{ inputs.test_type }} != "whl" ]; then
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl
fi
# install timm without dependencies
pip install --no-deps git+https://github.com/huggingface/pytorch-image-models@$TIMM_COMMIT_ID
# install timm dependencies without torch and torchvision
Expand Down
16 changes: 12 additions & 4 deletions .github/workflows/_linux_ut.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,11 @@ on:
type: string
default: 'linux.idc.xpu'
description: Runner label

test_type:
mengfei25 marked this conversation as resolved.
Show resolved Hide resolved
required: false
type: string
default: 'lts'
description: test type, lts/rolling/whl
permissions: read-all

jobs:
Expand Down Expand Up @@ -79,9 +83,13 @@ jobs:
else
export _GLIBCXX_USE_CXX11_ABI=1
fi
WERROR=1 python setup.py bdist_wheel
pip install --force-reinstall dist/*.whl
git clone https://github.com/pytorch/vision && cd vision && python setup.py install && cd ..
if [ ${{ inputs.test_type }} != "whl" ]; then
WERROR=1 python setup.py bdist_wheel
pip install --force-reinstall dist/*.whl
git clone https://github.com/pytorch/vision && cd vision && python setup.py install && cd ..
else
pip install torch torchvision torchaudio pytorch-triton-xpu --pre --index-url https://download.pytorch.org/whl/nightly/xpu
mengfei25 marked this conversation as resolved.
Show resolved Hide resolved
fi
pip install -r .ci/docker/requirements-ci.txt
- name: Torch Config
run: |
Expand Down
Loading