mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
a23ac6f8bd
Enable xpu runtime pypi packages as dependencies of XPU CD wheels both for Linux and Windows. Fixes https://github.com/pytorch/pytorch/issues/135867 Works for https://github.com/pytorch/pytorch/issues/139722 and https://github.com/pytorch/pytorch/issues/114850 Pull Request resolved: https://github.com/pytorch/pytorch/pull/141135 Approved by: https://github.com/atalman
19 lines
297 B
Bash
19 lines
297 B
Bash
#!/bin/bash
|
|
set -eux -o pipefail
|
|
|
|
source "${BINARY_ENV_FILE:-/c/w/env}"
|
|
|
|
export CUDA_VERSION="${DESIRED_CUDA/cu/}"
|
|
export VC_YEAR=2019
|
|
|
|
if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
|
|
export VC_YEAR=2022
|
|
export XPU_VERSION=2025.0
|
|
fi
|
|
|
|
pushd "$BUILDER_ROOT"
|
|
|
|
./windows/internal/smoke_test.bat
|
|
|
|
popd
|