mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
06c7516994
Including below changes, - Add XPU support package 2025.2 build and test in CI for both Linux and Windows - Keep XPU support package 2025.1 build in CI to ensure no break issue until PyTorch 2.9 release - Upgrade XPU support package from 2025.1 to 2025.2 in CD for both Linux and Windows - Rename Linux CI job name & image name to n & n-1 - Update XPU runtime pypi packages dependencies of CD wheels - Remove deprecated support package version docker image build Pull Request resolved: https://github.com/pytorch/pytorch/pull/158733 Approved by: https://github.com/EikanWang, https://github.com/atalman
39 lines
928 B
Bash
39 lines
928 B
Bash
#!/bin/bash
|
|
set -eux -o pipefail
|
|
|
|
source "${BINARY_ENV_FILE:-/c/w/env}"
|
|
mkdir -p "$PYTORCH_FINAL_PACKAGE_DIR"
|
|
|
|
if [[ "$OS" != "windows-arm64" ]]; then
|
|
export CUDA_VERSION="${DESIRED_CUDA/cu/}"
|
|
export USE_SCCACHE=1
|
|
export SCCACHE_BUCKET=ossci-compiler-cache
|
|
export SCCACHE_IGNORE_SERVER_IO_ERROR=1
|
|
export VC_YEAR=2022
|
|
fi
|
|
|
|
if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
|
|
export VC_YEAR=2022
|
|
export USE_SCCACHE=0
|
|
export XPU_VERSION=2025.2
|
|
fi
|
|
|
|
echo "Free space on filesystem before build:"
|
|
df -h
|
|
|
|
pushd "$PYTORCH_ROOT/.ci/pytorch/"
|
|
export NIGHTLIES_PYTORCH_ROOT="$PYTORCH_ROOT"
|
|
|
|
if [[ "$OS" == "windows-arm64" ]]; then
|
|
if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
|
|
./windows/arm64/build_libtorch.bat
|
|
elif [[ "$PACKAGE_TYPE" == 'wheel' ]]; then
|
|
./windows/arm64/build_pytorch.bat
|
|
fi
|
|
else
|
|
./windows/internal/build_wheels.bat
|
|
fi
|
|
|
|
echo "Free space on filesystem after build:"
|
|
df -h
|