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
Peter Hawkins dbf13252f0 Copybara import of the project: 
--
3905d6123bdc22f505934242363fda426c99c4cf by Peter Hawkins <phawkins@google.com>:

Update flatbuffers.

Use upstream flatbuffer bazel scripts, with a couple of small patches to fix:
* https://github.com/google/flatbuffers/issues/8087 (remove npm references)
* https://github.com/google/flatbuffers/pull/8088 (fix flatc build failure due to main() removal by linker)

COPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/17502 from hawkinsp:fb 3905d6123bdc22f505934242363fda426c99c4cf
PiperOrigin-RevId: 563543954
2023-09-07 14:27:25 -07:00
..
BUILD
Copybara import of the project:
2023-09-07 14:27:25 -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