mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 20:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lib
History
Skye Wanderman-Milne 7aa7e158f8 Modify JaxArrayTest.test_defragment to work on any numbers of devices 
Also skip it when the PJRT C API is enabled, since the C API only supports auto defrag.

PiperOrigin-RevId: 509933635
2023-02-15 14:36:03 -08:00
..
mlir
CI: fix mypy jaxlib version
2023-01-24 06:57:23 -08:00
__init__.py
Bump the minimum jaxlib version to 0.4.1.
2022-12-19 17:49:24 +00:00
xla_bridge.py
Modify JaxArrayTest.test_defragment to work on any numbers of devices
2023-02-15 14:36:03 -08:00