mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax_plugins
History
Peter Hawkins 3f91b4b43a Move jaxlib/{cuda,rocm}_plugin_extension into jaxlib/{cuda/rocm}/ 
Move the common jaxlib/gpu_plugin_extension into jaxlib/gpu/

Cleanup only, no functional changes intended.

PiperOrigin-RevId: 738183402
2025-03-18 16:29:37 -07:00
..
cuda
Move jaxlib/{cuda,rocm}_plugin_extension into jaxlib/{cuda/rocm}/
2025-03-18 16:29:37 -07:00
rocm
Move jaxlib/{cuda,rocm}_plugin_extension into jaxlib/{cuda/rocm}/
2025-03-18 16:29:37 -07:00
BUILD.bazel
Don't pass --nocheck_visibility to Bazel.
2024-09-27 11:19:42 -04:00