https://github.com/google/jax
History
Tip revision: 35172d2aaeb5a5a26fe2a95b24b439e19d3f5e2d authored by Christian Sigg on 04 January 2022, 21:43:29 UTC
[JAX] Avoid building arity-1 tuples in XLA lowering.
Tip revision: 35172d2
File Mode Size
mlir
BUILD -rw-r--r-- 13.5 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-- 27.6 KB
lapack_kernels.h -rw-r--r-- 6.0 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-- 925 bytes

back to top