diff --git a/.github/actions/inductor-xpu-e2e-test/action.yml b/.github/actions/inductor-xpu-e2e-test/action.yml index 4af472cce..0ded9afdd 100644 --- a/.github/actions/inductor-xpu-e2e-test/action.yml +++ b/.github/actions/inductor-xpu-e2e-test/action.yml @@ -48,14 +48,12 @@ runs: source activate e2e_ci source .github/scripts/env.sh if [[ ${{ inputs.suite }} == *"torchbench"* ]]; then - if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; 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 - 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 - fi + cd ../ && rm -rf audio && git clone --single-branch -b main https://github.com/pytorch/audio.git + cd audio && git checkout 97ed7b36b7a741253d4e41e4da3c901d83294503 + python setup.py bdist_wheel && pip uninstall torchaudio -y && pip install dist/*.whl + cd ../ && rm -rf vision && git clone --single-branch -b main https://github.com/pytorch/vision.git + cd vision && git checkout d23a6e1664d20707c11781299611436e1f0c104f + python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl 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 @@ -72,7 +70,7 @@ runs: if [[ ${{ inputs.suite }} == *"timm_models"* ]]; then if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then cd ../ && rm -rf vision && git clone --single-branch -b main https://github.com/pytorch/vision.git - cd vision && git checkout $TORCHVISION_COMMIT_ID + cd vision && git checkout d23a6e1664d20707c11781299611436e1f0c104f python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl fi # install timm without dependencies @@ -102,6 +100,8 @@ runs: } } set -xe + conda info -e + conda list for suite in $(echo ${{ inputs.suite }} |sed 's/,/ /g') do contains "huggingface,timm_models,torchbench" $suite diff --git a/.github/workflows/_linux_ut.yml b/.github/workflows/_linux_ut.yml index e004f355f..437c55c9e 100644 --- a/.github/workflows/_linux_ut.yml +++ b/.github/workflows/_linux_ut.yml @@ -103,7 +103,9 @@ jobs: 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 --pre --index-url https://download.pytorch.org/whl/nightly/xpu + # pip install torch torchvision torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/xpu + pip install torch --index-url https://download.pytorch.org/whl/test/xpu + git clone https://github.com/pytorch/vision && cd vision && git checkout d23a6e1664d20707c11781299611436e1f0c104f && python setup.py install && cd .. fi pip install -r .ci/docker/requirements-ci.txt - name: Torch Config diff --git a/.github/workflows/nightly_ondemand_whl.yml b/.github/workflows/nightly_ondemand_whl.yml index 641ca6865..3e95199b2 100644 --- a/.github/workflows/nightly_ondemand_whl.yml +++ b/.github/workflows/nightly_ondemand_whl.yml @@ -113,7 +113,8 @@ jobs: git status && git show -s pip install -r requirements.txt cd ../ - pip install torch torchvision torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/xpu + # pip install torch torchvision torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/xpu + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/xpu - name: Identify pinned versions id: pinned run: |