Revision 2c23c947a52331f6e4352bffc9bb185bc1c8cacd authored by jax authors on 16 March 2022, 19:55:19 UTC, committed by jax authors on 16 March 2022, 19:55:19 UTC
PiperOrigin-RevId: 435131902
File | Mode | Size |
---|---|---|
mlir | ||
BUILD | -rw-r--r-- | 19.0 KB |
cpu_feature_guard.c | -rw-r--r-- | 5.2 KB |
cpu_kernels.cc | -rw-r--r-- | 6.0 KB |
cublas.cc | -rw-r--r-- | 3.2 KB |
cublas_kernels.cc | -rw-r--r-- | 8.4 KB |
cublas_kernels.h | -rw-r--r-- | 1.7 KB |
cuda_gpu_kernel_helpers.cc | -rw-r--r-- | 4.9 KB |
cuda_gpu_kernel_helpers.h | -rw-r--r-- | 2.7 KB |
cuda_linalg.cc | -rw-r--r-- | 1.7 KB |
cuda_linalg.py | -rw-r--r-- | 2.1 KB |
cuda_lu_pivot_kernels.cc | -rw-r--r-- | 1.7 KB |
cuda_lu_pivot_kernels.cu.cc | -rw-r--r-- | 2.8 KB |
cuda_lu_pivot_kernels.h | -rw-r--r-- | 1.4 KB |
cuda_prng.cc | -rw-r--r-- | 1.3 KB |
cuda_prng.py | -rw-r--r-- | 1.8 KB |
cuda_prng_kernels.cc | -rw-r--r-- | 1.6 KB |
cuda_prng_kernels.cu.cc | -rw-r--r-- | 3.8 KB |
cuda_prng_kernels.h | -rw-r--r-- | 1.3 KB |
cusolver.cc | -rw-r--r-- | 19.2 KB |
cusolver.py | -rw-r--r-- | 14.7 KB |
cusolver_kernels.cc | -rw-r--r-- | 31.2 KB |
cusolver_kernels.h | -rw-r--r-- | 3.5 KB |
cusparse.cc | -rw-r--r-- | 23.1 KB |
cusparse.py | -rw-r--r-- | 11.7 KB |
cusparse_kernels.cc | -rw-r--r-- | 22.0 KB |
cusparse_kernels.h | -rw-r--r-- | 4.3 KB |
gpu_kernels.cc | -rw-r--r-- | 3.4 KB |
handle_pool.h | -rw-r--r-- | 3.3 KB |
hip_gpu_kernel_helpers.cc | -rw-r--r-- | 6.2 KB |
hip_gpu_kernel_helpers.h | -rw-r--r-- | 2.8 KB |
hip_linalg.cc | -rw-r--r-- | 1.7 KB |
hip_linalg.py | -rw-r--r-- | 2.0 KB |
hip_lu_pivot_kernels.cc | -rw-r--r-- | 1.7 KB |
hip_lu_pivot_kernels.h | -rw-r--r-- | 1.4 KB |
hip_lu_pivot_kernels.hip.cc | -rw-r--r-- | 2.8 KB |
hip_prng.cc | -rw-r--r-- | 1.3 KB |
hip_prng.py | -rw-r--r-- | 1.8 KB |
hip_prng_kernels.cc | -rw-r--r-- | 1.6 KB |
hip_prng_kernels.h | -rw-r--r-- | 1.2 KB |
hip_prng_kernels.hip.cc | -rw-r--r-- | 3.7 KB |
hipblas.cc | -rw-r--r-- | 3.2 KB |
hipblas_kernels.cc | -rw-r--r-- | 8.4 KB |
hipblas_kernels.h | -rw-r--r-- | 1.6 KB |
hipsolver.cc | -rw-r--r-- | 14.8 KB |
hipsolver.py | -rw-r--r-- | 13.8 KB |
hipsolver_kernels.cc | -rw-r--r-- | 22.4 KB |
hipsolver_kernels.h | -rw-r--r-- | 2.8 KB |
hipsparse.cc | -rw-r--r-- | 22.9 KB |
hipsparse.py | -rw-r--r-- | 11.8 KB |
hipsparse_kernels.cc | -rw-r--r-- | 20.6 KB |
hipsparse_kernels.h | -rw-r--r-- | 4.1 KB |
init.py | -rw-r--r-- | 630 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-- | 23.3 KB |
lapack_kernels.cc | -rw-r--r-- | 25.0 KB |
lapack_kernels.h | -rw-r--r-- | 6.2 KB |
lapack_kernels_using_lapack.cc | -rw-r--r-- | 3.9 KB |
pocketfft.cc | -rw-r--r-- | 1.0 KB |
pocketfft.fbs | -rw-r--r-- | 1009 bytes |
pocketfft.py | -rw-r--r-- | 5.5 KB |
pocketfft_kernels.cc | -rw-r--r-- | 3.7 KB |
pocketfft_kernels.h | -rw-r--r-- | 712 bytes |
setup.cfg | -rw-r--r-- | 38 bytes |
setup.py | -rw-r--r-- | 1.7 KB |
version.py | -rw-r--r-- | 924 bytes |
Computing file changes ...