Skip to content

Commit

Permalink
[CI] Test that NumPy-2.X builds are backward compatible with 1.X (#12…
Browse files Browse the repository at this point in the history
…3354)

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: #122157
Approved by: https://github.com/atalman

(cherry picked from commit 03b987f)

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
  • Loading branch information
pytorchbot and malfet committed Apr 4, 2024
1 parent 108305e commit dc81d19
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions .ci/pytorch/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -255,6 +255,11 @@ else
# 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
Expand Down

0 comments on commit dc81d19

Please sign in to comment.