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
Paweł Paruzel 5fc992e5e1 Determine LAPACK workspaces during SVD kernel runtime 
The SVD kernel implementation used to require workspace shapes to be determined prior to the custom call on the JAX's side. The new FFI kernels need not demand these shapes to be specified anymore. They are evaluated during kernel runtime.

PiperOrigin-RevId: 662413273
2024-08-13 01:17:44 -07:00
..
_lapack
Determine LAPACK workspaces during SVD kernel runtime
2024-08-13 01:17:44 -07:00
BUILD
Determine LAPACK workspaces during SVD kernel runtime
2024-08-13 01:17:44 -07:00
cpu_kernels.cc
Port Eigenvalue Decompositions to XLA's FFI
2024-08-05 03:18:13 -07:00
lapack_kernels_using_lapack.cc
Port Eigenvalue Decompositions to XLA's FFI
2024-08-05 03:18:13 -07:00
lapack_kernels.cc
Determine LAPACK workspaces during SVD kernel runtime
2024-08-13 01:17:44 -07:00
lapack_kernels.h
Determine LAPACK workspaces during SVD kernel runtime
2024-08-13 01:17:44 -07:00
lapack.cc
Determine LAPACK workspaces during SVD kernel runtime
2024-08-13 01:17:44 -07:00