frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI] Update XPU ci test python version to 3.9 (#134214)

Browse Source 
Works for https://github.com/pytorch/pytorch/issues/114850
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134214
Approved by: https://github.com/EikanWang, https://github.com/malfet
This commit is contained in:
chuanqiw
2024-08-24 18:11:36 +00:00
committed by PyTorch MergeBot
parent a63efee5cd
commit 6245d5b87b
3 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.ci/docker/build.sh
View File

@ -308,7 +308,7 @@ case "$image" in
TRITON=yes
;;
pytorch-linux-jammy-xpu-2024.0-py3)
ANACONDA_PYTHON_VERSION=3.8
ANACONDA_PYTHON_VERSION=3.9
GCC_VERSION=11
PROTOBUF=yes
DB=yes

11
.ci/docker/common/install_conda.sh
View File

@ -78,19 +78,20 @@ fi
CONDA_COMMON_DEPS="astunparse pyyaml setuptools openblas==0.3.25=*openmp* ninja==1.11.1 scons==4.5.2"
if [ "$ANACONDA_PYTHON_VERSION" = "3.8" ]; then
conda_install numpy=1.24.4 ${CONDA_COMMON_DEPS}
NUMPY_VERSION=1.24.4
else
conda_install numpy=1.26.2 ${CONDA_COMMON_DEPS}
NUMPY_VERSION=1.26.2
fi
else
CONDA_COMMON_DEPS="astunparse pyyaml mkl=2021.4.0 mkl-include=2021.4.0 setuptools"
if [ "$ANACONDA_PYTHON_VERSION" = "3.11" ] || [ "$ANACONDA_PYTHON_VERSION" = "3.12" ] || [ "$ANACONDA_PYTHON_VERSION" = "3.13" ]; then
conda_install numpy=1.26.0 ${CONDA_COMMON_DEPS}
NUMPY_VERSION=1.26.0
else
conda_install numpy=1.21.2 ${CONDA_COMMON_DEPS}
NUMPY_VERSION=1.21.2
fi
fi
conda_install ${CONDA_COMMON_DEPS}
# Install llvm-8 as it is required to compile llvmlite-0.30.0 from source
# and libpython-static for torch deploy
@ -112,7 +113,7 @@ fi
# Install some other packages, including those needed for Python test reporting
pip_install -r /opt/conda/requirements-ci.txt
pip_install numpy=="$NUMPY_VERSION"
pip_install -U scikit-learn
if [ -n "$DOCS" ]; then

18
.github/workflows/xpu.yml vendored
View File

@ -21,13 +21,13 @@ jobs:
curr_branch: ${{ github.head_ref || github.ref_name }}
curr_ref_type: ${{ github.ref_type }}
linux-jammy-xpu-py3_8-build:
name: linux-jammy-xpu-py3.8
linux-jammy-xpu-py3_9-build:
name: linux-jammy-xpu-py3.9
uses: ./.github/workflows/_linux-build.yml
needs: get-label-type
with:
runner_prefix: ${{ needs.get-label-type.outputs.label-type }}
build-environment: linux-jammy-xpu-py3.8
build-environment: linux-jammy-xpu-py3.9
docker-image-name: pytorch-linux-jammy-xpu-2024.0-py3
runner: linux.12xlarge
test-matrix: |
@ -38,14 +38,14 @@ jobs:
{ config: "default", shard: 4, num_shards: 4, runner: "linux.idc.xpu" },
]}
linux-jammy-xpu-py3_8-test:
name: linux-jammy-xpu-py3.8
linux-jammy-xpu-py3_9-test:
name: linux-jammy-xpu-py3.9
uses: ./.github/workflows/_xpu-test.yml
needs: linux-jammy-xpu-py3_8-build
needs: linux-jammy-xpu-py3_9-build
permissions:
id-token: write
contents: read
with:
build-environment: linux-jammy-xpu-py3.8
docker-image: ${{ needs.linux-jammy-xpu-py3_8-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-xpu-py3_8-build.outputs.test-matrix }}
build-environment: linux-jammy-xpu-py3.9
docker-image: ${{ needs.linux-jammy-xpu-py3_9-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-xpu-py3_9-build.outputs.test-matrix }}