Revision f47ce791da7ef522405fdda189d7ace827530573 authored by jax authors on 15 March 2022, 21:56:00 UTC, committed by jax authors on 15 March 2022, 21:56:00 UTC
PiperOrigin-RevId: 434863868
| 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 ...
