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
Antonio Sanchez 873cffc776 Use TSL's import of DUCC. 
This is a necessary first step before adding DUCC support to XLA,
otherwise the JAX tests in the XLA repo pull from JAX's copy,
which has slightly different build rules.

PiperOrigin-RevId: 576880208
2023-10-26 08:26:56 -07:00
..
BUILD
Use TSL's import of DUCC.
2023-10-26 08:26:56 -07:00
cpu_kernels.cc
Add support for dynamic shapes to jax.fft.
2023-06-17 04:50:54 -07:00
ducc_fft_kernels.cc
Update ducc to commit: 2b2cead005e08d2632478e831d7f45da754162dc
2023-08-07 13:06:43 -07:00
ducc_fft_kernels.h
Add support for dynamic shapes to jax.fft.
2023-06-17 04:50:54 -07:00
ducc_fft.cc
Switch jaxlib to use nanobind instead of pybind11.
2023-08-24 16:07:56 -07:00
ducc_fft.fbs
Add support for dynamic shapes to jax.fft.
2023-06-17 04:50:54 -07:00
lapack_kernels_using_lapack.cc
Add support for Hessenberg and tridiagonal matrix reductions on CPU.
2022-11-09 06:23:55 -08:00
lapack_kernels.cc
Add support for Hessenberg and tridiagonal matrix reductions on CPU.
2022-11-09 06:23:55 -08:00
lapack_kernels.h
Switch JAX to use the OpenXLA repository.
2023-03-13 18:38:26 +00:00
lapack.cc
Switch jaxlib to use nanobind instead of pybind11.
2023-08-24 16:07:56 -07:00