Revision ecb27a9b244a609b779810444b3a109322b96963 authored by Yash Katariya on 23 September 2022, 18:40:01 UTC, committed by jax authors on 23 September 2022, 18:40:32 UTC
PiperOrigin-RevId: 476422732
1 parent d078f3f
File | Mode | Size |
---|---|---|
cuda | ||
mlir | ||
rocm | ||
BUILD | -rw-r--r-- | 5.5 KB |
README.md | -rw-r--r-- | 402 bytes |
cpu_feature_guard.c | -rw-r--r-- | 5.2 KB |
cpu_kernels.cc | -rw-r--r-- | 6.0 KB |
ducc_fft.cc | -rw-r--r-- | 2.4 KB |
ducc_fft.fbs | -rw-r--r-- | 1003 bytes |
ducc_fft.py | -rw-r--r-- | 4.5 KB |
ducc_fft_kernels.cc | -rw-r--r-- | 4.3 KB |
ducc_fft_kernels.h | -rw-r--r-- | 803 bytes |
gpu_linalg.py | -rw-r--r-- | 2.3 KB |
gpu_prng.py | -rw-r--r-- | 2.1 KB |
gpu_solver.py | -rw-r--r-- | 15.9 KB |
gpu_sparse.py | -rw-r--r-- | 12.1 KB |
handle_pool.h | -rw-r--r-- | 3.3 KB |
init.py | -rw-r--r-- | 629 bytes |
jax.bzl | -rw-r--r-- | 7.1 KB |
kernel_helpers.h | -rw-r--r-- | 1.7 KB |
kernel_pybind11_helpers.h | -rw-r--r-- | 1.7 KB |
lapack.cc | -rw-r--r-- | 9.6 KB |
lapack.py | -rw-r--r-- | 19.1 KB |
lapack_kernels.cc | -rw-r--r-- | 25.4 KB |
lapack_kernels.h | -rw-r--r-- | 6.6 KB |
lapack_kernels_using_lapack.cc | -rw-r--r-- | 3.9 KB |
mhlo_helpers.py | -rw-r--r-- | 2.4 KB |
setup.cfg | -rw-r--r-- | 38 bytes |
setup.py | -rw-r--r-- | 2.4 KB |
symlink_files.bzl | -rw-r--r-- | 6.0 KB |
Computing file changes ...