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

1 Commits

Author SHA1 Message Date
Dan Foreman-Mackey
f20efc630f Move jaxlib GPU handlers to separate build target. 
In anticipation of refactoring the jaxlib GPU custom calls into FFI calls, this change moves the implementation of `BlasHandlePool`, `SolverHandlePool`, and `SpSolverHandlePool` into new target.

PiperOrigin-RevId: 658497960
 2024-08-01 12:30:04 -07:00