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
George Necula 6bfbb4593a Remove old ducc_fft custom call. 
Starting in June 2023 we have switched the CPU lowering for FFT to use
the new custom call dynamic_ducc_fft. We are now out of the backwards
compatibility window and we remove the old ducc_fft.

We need to keep dynamic_ducc_fft a little bit longer (May 2024).

PiperOrigin-RevId: 627981921
2024-04-25 00:29:11 -07:00
..
_ducc_fft.pyi
Add python stub files for jaxlib/cpu C++ Python extensions.
2023-11-28 08:45:24 -08:00
_lapack.pyi
Add nb::arg to nanobind definitions to generate better python annotations.
2023-11-30 10:39:28 -08:00
BUILD
Merge pull request #19288 from pearu:pearu/int32-overflow
2024-02-20 12:43:16 -08:00
cpu_kernels.cc
Add support for dynamic shapes to jax.fft.
2023-06-17 04:50:54 -07:00
ducc_fft_kernels.cc
Remove old ducc_fft custom call.
2024-04-25 00:29:11 -07:00
ducc_fft_kernels.h
Remove old ducc_fft custom call.
2024-04-25 00:29:11 -07:00
ducc_fft.cc
Remove old ducc_fft custom call.
2024-04-25 00:29:11 -07:00
ducc_fft.fbs
Remove old ducc_fft custom call.
2024-04-25 00:29:11 -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
Prevent silent overflow in lapack worker size calculations.
2024-02-20 11:04:06 +02:00
lapack_kernels.h
Switch JAX to use the OpenXLA repository.
2023-03-13 18:38:26 +00:00
lapack.cc
Add nb::arg to nanobind definitions to generate better python annotations.
2023-11-30 10:39:28 -08:00