mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26: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-05-29 08:48:49 -07:00
parent cc0a20f4b1
commit 6fb357d651
2 changed files with 17 additions and 6 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: 'e38ce3466e596ed2b8fa4638f161f5563ded81a8' # Latest commit as of 2024-04-15
ref: '33f2d2ea2f3dd2b3ceeeb4519d55e08096184149' # Latest commit as of 2024-05-28
submodules: 'true'
path: 'array-api-tests'
- name: Set up Python ${{ matrix.python-version }}

21
jax/experimental/array_api/skips.txt
View File

@ -9,16 +9,27 @@ array_api_tests/test_array_object.py::test_setitem
array_api_tests/test_special_cases.py::test_binary
array_api_tests/test_special_cases.py::test_unary
# fft test suite is buggy as of 83f0bcdc
array_api_tests/test_fft.py
# 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
array_api_tests/test_linalg.py::test_trace
# 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]
array_api_tests/test_signatures.py::test_func_signature[logical_xor]
# Returns int32 when int64 is expected
array_api_tests/test_searching_functions.py::test_searchsorted
# Various info functions not yet defined
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]