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 |
|