Revision 34b3a7f106baa9099802f246fda6ad3476f32d26 authored by Qiao Zhang on 16 February 2022, 19:26:23 UTC, committed by jax authors on 16 February 2022, 19:41:41 UTC
PiperOrigin-RevId: 429097675
1 parent e1fd630
| File | Mode | Size |
|---|---|---|
| mlir | ||
| BUILD | -rw-r--r-- | 13.7 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.4 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 |
| 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 |
| rocblas.cc | -rw-r--r-- | 35.3 KB |
| rocm_gpu_kernel_helpers.cc | -rw-r--r-- | 1.5 KB |
| rocm_gpu_kernel_helpers.h | -rw-r--r-- | 1.7 KB |
| rocsolver.py | -rw-r--r-- | 13.3 KB |
| 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 ...
