[CI] Test that NumPy-2.X builds are backward compatible with 1.X (#122157)
By compiling PyTorch against 2.x RC, but running all the tests with Numpy-1.X
This has no affects on binary builds
Pull Request resolved: https://github.com/pytorch/pytorch/pull/122157
Approved by: https://github.com/atalman
diff --git a/.ci/pytorch/build.sh b/.ci/pytorch/build.sh
index fe72f8b..9e3ee01 100755
--- a/.ci/pytorch/build.sh
+++ b/.ci/pytorch/build.sh
@@ -255,6 +255,11 @@
# or building non-XLA tests.
if [[ "$BUILD_ENVIRONMENT" != *rocm* &&
"$BUILD_ENVIRONMENT" != *xla* ]]; then
+ if [[ "$BUILD_ENVIRONMENT" != *py3.8* ]]; then
+ # Install numpy-2.0 release candidate for builds
+ # Which should be backward compatible with Numpy-1.X
+ python -mpip install --pre numpy==2.0.0b1
+ fi
WERROR=1 python setup.py bdist_wheel
else
python setup.py bdist_wheel
