mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 06:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Skye Wanderman-Milne 7c2454e969 Update jaxlib version, minimum jaxlib version, readme, and changelog. 
Bumping the min jaxlib version to support https://github.com/google/jax/pull/5213.
2021-01-15 12:56:08 -08:00
..
BUILD
Merge pull request #5115 from inailuig:rocm-gpukernels
2020-12-17 14:01:04 -08:00
cublas.cc
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
cuda_gpu_kernel_helpers.cc
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
cuda_gpu_kernel_helpers.h
Fix header guard typo.
2020-12-23 11:25:23 -08: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 BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01: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
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
cusolver.cc
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01:00
cusolver.py
Add BLAS and LAPACK gpu kernels for ROCm
2020-12-16 16:00:17 +01: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
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
[XLA:Python] Validate shapes in Python bindings to avoid crashes.
2020-10-15 13:10:04 -07: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
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -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
Change jaxlib build rules to build a wheel, rather than writing output to the source directory.
2020-11-20 11:47:00 -05:00
version.py
Update jaxlib version, minimum jaxlib version, readme, and changelog.
2021-01-15 12:56:08 -08:00