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 33a9db3943 Move FFI helper macros from jaxlib/cpu/lapack_kernels.cc to a jaxlib/ffi_helpers.h. 
Some of the macros that were used in jaxlib's FFI calls to LAPACK turned out to
be useful for other FFI calls. This change consolidates these macros in the
ffi_helper header.

PiperOrigin-RevId: 651166306
2024-07-10 15:09:45 -07:00
..
_lapack
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
BUILD
Move FFI helper macros from jaxlib/cpu/lapack_kernels.cc to a jaxlib/ffi_helpers.h.
2024-07-10 15:09:45 -07:00
cpu_kernels.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack_kernels_using_lapack.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack_kernels.cc
Move FFI helper macros from jaxlib/cpu/lapack_kernels.cc to a jaxlib/ffi_helpers.h.
2024-07-10 15:09:45 -07:00
lapack_kernels.h
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack.h
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00