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

[array API] update test suite to most recent commit

Browse Source
This commit is contained in:
Jake VanderPlas 2024-08-08 12:33:30 -07:00
parent e6303244bf
commit d999208863
3 changed files with 6 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -25,7 +25,7 @@ jobs:
with:
repository: data-apis/array-api-tests
# TODO(jakevdp) update this to a stable release/tag when available.
ref: '33f2d2ea2f3dd2b3ceeeb4519d55e08096184149' # Latest commit as of 2024-05-28
ref: 'db95e67b29235249e5776ca2b6bb4e77117e0690' # Latest commit as of 2024-08-08
submodules: 'true'
path: 'array-api-tests'
- name: Set up Python ${{ matrix.python-version }}

3
pyproject.toml
View File

@ -60,7 +60,8 @@ filterwarnings = [
# TODO(jakevdp): remove when array_api_tests stabilize
"default:.*not machine-readable.*:UserWarning",
"default:Special cases found for .* but none were parsed.*:UserWarning",
"default:.*is not JSON-serializable. Using the repr instead.",
"default:.*is not JSON-serializable. Using the repr instead.*:UserWarning",
"default:The .* method is good for exploring strategies.*",
# These are transitive warnings coming from TensorFlow dependencies.
# TODO(slebedev): Remove once we bump the minimum TensorFlow version.

31
tests/array_api_skips.txt
View File

@ -4,36 +4,11 @@
array_api_tests/test_data_type_functions.py::test_finfo[float32]
# Test suite attempts in-place mutation:
array_api_tests/test_special_cases.py::test_iop
array_api_tests/test_special_cases.py::test_nan_propagation
array_api_tests/test_array_object.py::test_setitem
array_api_tests/test_array_object.py::test_setitem_masking
# Raises NonInteractiveExampleWarning
array_api_tests/test_special_cases.py::test_binary
array_api_tests/test_special_cases.py::test_unary
# Pending implementation update for proper dtype promotion behavior,
# see https://github.com/data-apis/array-api-tests/issues/234
array_api_tests/test_statistical_functions.py::test_sum
array_api_tests/test_statistical_functions.py::test_prod
# Pending bugfix, see https://github.com/data-apis/array-api-tests/issues/256
array_api_tests/test_signatures.py::test_func_signature[logical_and]
array_api_tests/test_signatures.py::test_func_signature[logical_or]
array_api_tests/test_signatures.py::test_func_signature[logical_xor]
# Returns wrong zero sign
array_api_tests/test_special_cases.py::test_unary[sign((x_i is -0 or x_i == +0)) -> 0]
# Returns int32 when int64 is expected
array_api_tests/test_searching_functions.py::test_searchsorted
# Various info functions not yet defined
# Pending bugfix, see https://github.com/data-apis/array-api-tests/pull/262
array_api_tests/test_has_names.py::test_has_names[info-capabilities]
array_api_tests/test_has_names.py::test_has_names[info-default_device]
array_api_tests/test_has_names.py::test_has_names[info-default_dtypes]
array_api_tests/test_has_names.py::test_has_names[info-devices]
array_api_tests/test_has_names.py::test_has_names[info-dtypes]
array_api_tests/test_signatures.py::test_func_signature[capabilities]
array_api_tests/test_signatures.py::test_func_signature[default_device]
array_api_tests/test_signatures.py::test_func_signature[default_dtypes]
array_api_tests/test_signatures.py::test_func_signature[devices]
array_api_tests/test_signatures.py::test_func_signature[dtypes]