mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 21:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,928 Commits 508 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Andrey Portnoy
ec5c4f5a10 Add CUDA custom call example as a JAX test 2024-06-17 15:21:49 -04:00
Yash Katariya
395d3cb79e Bump minimum jaxlib version to 0.4.27 
xla_extension_version is 261 and mlir_api_version is 56

PiperOrigin-RevId: 631579739
 2024-05-07 16:07:59 -07:00
jax authors
7ecd22cfcc Exclude test_gpu_memory_allocation from pytest execution. 
PiperOrigin-RevId: 592664477
 2023-12-20 14:53:33 -08:00
Peter Hawkins
67d5c3bdea [JAX:GPU] Add a test that verifies that the XLA_PYTHON_CLIENT_PREALLOCATE environment variable is parsed correctly. 
Fixes https://github.com/google/jax/issues/19035

PiperOrigin-RevId: 592322040
 2023-12-19 13:06:08 -08:00