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 98b87540a7 Avoid throwing exceptions in LAPACK CPU kernels. 
When an FFI kernel is executed, there isn't any global try/except block (I think!) so it's probably a good idea to avoid throwing.
Instead, it should be safer to handle mapping failures to ffi::Error manually.

PiperOrigin-RevId: 647348889
2024-06-27 09:41:07 -07:00
..
_lapack.pyi
Add nb::arg to nanobind definitions to generate better python annotations.
2023-11-30 10:39:28 -08:00
BUILD
Avoid throwing exceptions in LAPACK CPU kernels.
2024-06-27 09:41:07 -07:00
cpu_kernels.cc
Port LU Decomposition to XLA's FFI
2024-06-19 17:31:25 -07:00
lapack_kernels_using_lapack.cc
Port LU Decomposition to XLA's FFI
2024-06-19 17:31:25 -07:00
lapack_kernels.cc
Avoid throwing exceptions in LAPACK CPU kernels.
2024-06-27 09:41:07 -07:00
lapack_kernels.h
Port LU Decomposition to XLA's FFI
2024-06-19 17:31:25 -07:00
lapack.cc
Port LU Decomposition to XLA's FFI
2024-06-19 17:31:25 -07:00
lapack.h
Port LU Decomposition to XLA's FFI
2024-06-19 17:31:25 -07:00