mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cpu
History
Dan Foreman-Mackey ff4e0b1214 Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors. 
When linking the jaxlib `cpu_kernels` target and importing JAX, we currently silently fail to instantiate the CPU backend. This refactor means that we only ever define one version of the handlers.

PiperOrigin-RevId: 657186057
2024-07-29 06:59:44 -07:00
..
_lapack
Port Householder Product to XLA's FFI
2024-07-12 01:36:41 -07:00
BUILD
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.
2024-07-29 06:59:44 -07:00
cpu_kernels.cc
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.
2024-07-29 06:59:44 -07:00
lapack_kernels_using_lapack.cc
Port Triangular Solve to XLA's FFI
2024-07-24 02:15:41 -07:00
lapack_kernels.cc
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.
2024-07-29 06:59:44 -07:00
lapack_kernels.h
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.
2024-07-29 06:59:44 -07:00
lapack.cc
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.
2024-07-29 06:59:44 -07:00