|
..
|
|
mlir
|
|
BUILD
|
|
cpu_feature_guard.c
|
|
cpu_kernels.cc
|
|
cublas.cc
|
|
cublas_kernels.cc
|
|
cublas_kernels.h
|
|
cuda_gpu_kernel_helpers.cc
|
|
cuda_gpu_kernel_helpers.h
|
|
cuda_linalg.cc
|
|
cuda_linalg.py
|
|
cuda_lu_pivot_kernels.cc
|
|
cuda_lu_pivot_kernels.cu.cc
|
|
cuda_lu_pivot_kernels.h
|
|
cuda_prng.cc
|
|
cuda_prng.py
|
|
cuda_prng_kernels.cc
|
|
cuda_prng_kernels.cu.cc
|
|
cuda_prng_kernels.h
|
|
cusolver.cc
|
|
cusolver.py
|
|
cusolver_kernels.cc
|
|
cusolver_kernels.h
|
|
cusparse.cc
|
|
cusparse.py
|
|
cusparse_kernels.cc
|
|
cusparse_kernels.h
|
|
gpu_kernels.cc
|
|
handle_pool.h
|
|
init.py
|
|
jax.bzl
|
|
kernel_helpers.h
|
|
kernel_pybind11_helpers.h
|
|
lapack.cc
|
|
lapack.py
|
|
lapack_kernels.cc
|
|
lapack_kernels.h
|
|
pocketfft.cc
|
|
pocketfft.fbs
|
|
pocketfft.py
|
|
pocketfft_kernels.cc
|
|
pocketfft_kernels.h
|
|
rocblas.cc
|
|
rocm_gpu_kernel_helpers.cc
|
|
rocm_gpu_kernel_helpers.h
|
|
rocsolver.py
|
|
setup.cfg
|
|
setup.py
|
|
version.py
|