mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
jax authors 6aa20d8f8f Merge pull request #7294 from hawkinsp:py36 
PiperOrigin-RevId: 384994957
2021-07-15 13:19:23 -07:00
..
BUILD
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cpu_feature_guard.c
move #endif so that Windows doesn't have GetXCR0EAX defined twice
2021-06-04 17:26:27 -07:00
cublas.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cuda_gpu_kernel_helpers.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cuda_gpu_kernel_helpers.h
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cuda_linalg.py
Reimplement as JAX Primitive
2021-04-03 14:11:36 -07:00
cuda_lu_pivot_kernels.cc
Reimplement as JAX Primitive
2021-04-03 14:11:36 -07:00
cuda_lu_pivot_kernels.cu.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cuda_lu_pivot_kernels.h
Reimplement as JAX Primitive
2021-04-03 14:11:36 -07:00
cuda_prng_kernels.cc
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng_kernels.cu.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cuda_prng_kernels.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng.py
Small cleanups to dependency structure.
2021-01-20 12:43:28 -08:00
cusolver.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cusolver.py
Small cleanups to dependency structure.
2021-01-20 12:43:28 -08:00
cusparse.cc
Add source location and expression to error messages for CUDA API calls.
2021-07-15 15:42:46 +00:00
cusparse.py
Add lax.linalg.tridiagonal_solve(..), lowering to cusparse_gtsv2<T>() on GPU.
2021-06-02 13:49:02 +00:00
handle_pool.h
extract the shared handle pool code from cublas and cusolver
2020-12-16 16:00:16 +01:00
init.py
Fix build.py to work on Linux once again.
2020-11-19 13:14:34 -05:00
jax.bzl
Remove jax/BUILD file.
2021-05-19 12:55:34 -07:00
kernel_helpers.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
kernel_pybind11_helpers.h
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
lapack.pyx
Initialize variables in LAPACK work size queries to prevent false positives in memory sanitizers.
2021-02-01 16:38:24 -08:00
pocketfft.cc
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
pocketfft.fbs
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
pocketfft.py
Update flatbuffer ver2 EndVector usage.
2021-05-10 16:45:19 -07:00
rocblas.cc
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
rocm_gpu_kernel_helpers.cc
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
rocm_gpu_kernel_helpers.h
fix header guard typo
2021-01-13 15:55:14 +01:00
rocsolver.py
fix svd on ROCm
2021-01-13 15:54:21 +01:00
setup.cfg
Export LICENSE.txt file in jaxlib wheels.
2020-12-11 10:20:03 -05:00
setup.py
Drop Python 3.6 support.
2021-07-15 14:20:29 -04:00
version.py
Update changelog for jaxlib-0.1.69.
2021-07-12 12:06:41 -07:00