mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cpu
History
Peter Hawkins 91ffb640a8 Use thread-safe initialization of LAPACK kernels. 
Use absl::call_once instead of a GIL-protected global initialization.

In passing, also remove an unused function.

PiperOrigin-RevId: 714892175
2025-01-13 02:51:38 -08:00
..
_lapack
Port Schur Decomposition to XLA's FFI
2024-10-14 06:46:42 -07:00
BUILD
Use thread-safe initialization of LAPACK kernels.
2025-01-13 02:51:38 -08:00
cpu_kernels.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels_using_lapack.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels.h
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack.cc
Use thread-safe initialization of LAPACK kernels.
2025-01-13 02:51:38 -08:00