Description

honno

opened on Feb 20, 2024

Issue body actions

• Firstly we really need to
  • Replace numpy.array_api with array-api-strict
    • numpy.array_api is going to be removed and supplanted by this third-party library
    • As bought up in More improvements to test_linalg #101 (comment)
  • Also cover meta/ tests (we never did before!)
    • Would require one or two fixes of those tests.
  • Cover different ARRAY_API_TESTS_VERSION values too
    • Support for older versions of the standard array-api-strict#8 would be handy here
• Then ideally we're testing libraries covered by array-api-compat
  • NumPy
  • PyTorch
  • Dask
  • CuPy is probably a no-go because GPUs on CI is too finnicky
• I think it'd be really nice to test jax.experimental.array_api too as it has some distinct properties that we want to be mindful with for future test suite development.

Activity

1. honno

   added

   tracking

   on Feb 20, 2024
2. asmeurer commented on Mar 21, 2024

   

   asmeurer

   on Mar 21, 2024

   Member

   More actions

   For array-api-compat I have a script that I can run on a CUDA machine that runs the CuPy tests. We could have something like that here. We could also do that for PyTorch GPU. We would then just have to run it ourselves manually from time to time.

   Reacted by Matthew Barber

Issue actions