mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

array API: improve test coverage

Browse Source
This commit is contained in:
Jake VanderPlas 2024-12-05 12:56:07 -08:00
parent f73fa7a7ad
commit d88ef23a63
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/jax-array-api.yml vendored
View File

@ -38,11 +38,11 @@ jobs:
- name: Install dependencies
run: |
python -m pip install .[ci]
python -m pip install -r array-api-tests/requirements.txt
python -m pip install pytest-xdist -r array-api-tests/requirements.txt
- name: Run the test suite
env:
ARRAY_API_TESTS_MODULE: jax.numpy
JAX_ENABLE_X64: 'true'
run: |
cd ${GITHUB_WORKSPACE}/array-api-tests
pytest array_api_tests --max-examples=5 --derandomize --disable-deadline --skips-file ${GITHUB_WORKSPACE}/tests/array_api_skips.txt
pytest -n auto array_api_tests --derandomize --disable-deadline --skips-file ${GITHUB_WORKSPACE}/tests/array_api_skips.txt

6
tests/array_api_skips.txt
View File

@ -13,3 +13,9 @@ array_api_tests/test_special_cases.py::test_unary[sign((x_i is -0 or x_i == +0))
# Returns int32 when int64 is expected
array_api_tests/test_searching_functions.py::test_searchsorted
# clip out dtype has ambiguous semantics (https://github.com/numpy/numpy/issues/24976)
array_api_tests/test_operators_and_elementwise_functions.py::test_clip
# JAX raises a ValueError rather than the expected IndexError for out-of-bound axis
array_api_tests/test_manipulation_functions.py::test_expand_dims