diff options
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 2 | ||||
| -rw-r--r-- | tools/wheels/cibw_test_command.sh | 4 |
2 files changed, 5 insertions, 1 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index e27a39c3a..5dba5c68a 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -6721,7 +6721,7 @@ class TestDot: @pytest.mark.slow @pytest.mark.parametrize("dtype", [np.float64, np.complex128]) - @requires_memory(free_bytes=9*10**9) # complex case needs 8GiB+ + @requires_memory(free_bytes=18e9) # complex case needs 18GiB+ def test_huge_vectordot(self, dtype): # Large vector multiplications are chunked with 32bit BLAS # Test that the chunking does the right thing, see also gh-22262 diff --git a/tools/wheels/cibw_test_command.sh b/tools/wheels/cibw_test_command.sh index 92252d9cd..cbc735fc0 100644 --- a/tools/wheels/cibw_test_command.sh +++ b/tools/wheels/cibw_test_command.sh @@ -11,6 +11,10 @@ if [[ $RUNNER_OS == "Windows" ]]; then PY_DIR=$(python -c "import sys; print(sys.prefix)") mkdir $PY_DIR/libs fi + +# Set available memory value to avoid OOM problems on aarch64. +# See gh-22418. +export NPY_AVAILABLE_MEM="4 GB" python -c "import sys; import numpy; sys.exit(not numpy.test('full'))" python $PROJECT_DIR/tools/wheels/check_license.py |