File | Mode | Size |
---|---|---|
cuda | ||
mlir | ||
rocm | ||
BUILD | -rw-r--r-- | 4.6 KB |
cpu_feature_guard.c | -rw-r--r-- | 5.2 KB |
cpu_kernels.cc | -rw-r--r-- | 6.0 KB |
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.1 KB |
gpu_sparse.py | -rw-r--r-- | 11.3 KB |
handle_pool.h | -rw-r--r-- | 3.3 KB |
init.py | -rw-r--r-- | 623 bytes |
jax.bzl | -rw-r--r-- | 3.7 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-- | 20.5 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 |
pocketfft.cc | -rw-r--r-- | 1.0 KB |
pocketfft.fbs | -rw-r--r-- | 1009 bytes |
pocketfft.py | -rw-r--r-- | 7.7 KB |
pocketfft_kernels.cc | -rw-r--r-- | 3.8 KB |
pocketfft_kernels.h | -rw-r--r-- | 799 bytes |
setup.cfg | -rw-r--r-- | 38 bytes |
setup.py | -rw-r--r-- | 2.1 KB |
version.py | -rw-r--r-- | 925 bytes |