mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-06 00:54:56 +08:00
179e7d8624
Fixes #145702 Add `/d2implyavx512upperregs-` to disable compiler over-aggressive optimization, which caused involeved AVX512 register on AVX2 machine. Reference to: https://github.com/pytorch/pytorch/issues/145702#issuecomment-2874029459 Local test passed: <img width="1208" alt="image" src="https://github.com/user-attachments/assets/26f4cb91-6bb5-416f-aa35-c899eb1489b2" /> Pull Request resolved: https://github.com/pytorch/pytorch/pull/153480 Approved by: https://github.com/Blackhex, https://github.com/cyyever, https://github.com/atalman
23 lines
397 B
Bash
23 lines
397 B
Bash
#!/bin/bash
|
|
set -eux -o pipefail
|
|
|
|
source "${BINARY_ENV_FILE:-/c/w/env}"
|
|
|
|
export CUDA_VERSION="${DESIRED_CUDA/cu/}"
|
|
export VC_YEAR=2022
|
|
|
|
if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
|
|
export VC_YEAR=2022
|
|
export XPU_VERSION=2025.1
|
|
fi
|
|
|
|
pushd "$PYTORCH_ROOT/.ci/pytorch/"
|
|
|
|
if [[ "$OS" == "windows-arm64" ]]; then
|
|
./windows/arm64/smoke_test.bat
|
|
else
|
|
./windows/internal/smoke_test.bat
|
|
fi
|
|
|
|
popd
|