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 3d39b6e752 Port Cholesky Factorization to XLA's FFI 
This CL only contains the C++ changes. Python lowering code will be added after the forward compatibility window of 3 weeks.

PiperOrigin-RevId: 642954763
2024-06-13 05:44:36 -07:00
..
_lapack.pyi
Add nb::arg to nanobind definitions to generate better python annotations.
2023-11-30 10:39:28 -08:00
BUILD
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
cpu_kernels.cc
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
lapack_kernels_using_lapack.cc
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
lapack_kernels.cc
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
lapack_kernels.h
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
lapack.cc
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00
lapack.h
Port Cholesky Factorization to XLA's FFI
2024-06-13 05:44:36 -07:00