Skip to main content

Browse the archive

https://github.com/google/jax
22 May 2024, 15:31:03 UTC
History
Tip revision: 1189c3c62f39b6ba0ec464c21201bcb5b5041238 authored by jax authors on 15 April 2022, 19:28:57 UTC
Merge pull request #10312 from hawkinsp:jaxlib
Tip revision: 1189c3c
File Mode Size
mlir
BUILD -rw-r--r-- 19.1 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-- 3.4 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-- 3.1 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-- 32.1 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-- 26.4 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-- 3.4 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-- 3.1 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-- 28.4 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-- 26.3 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-- 51.7 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
pocketfft.cc -rw-r--r-- 1.0 KB
pocketfft.fbs -rw-r--r-- 1009 bytes
pocketfft.py -rw-r--r-- 8.0 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-- 1.7 KB
version.py -rw-r--r-- 924 bytes
back to top