mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 13:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Peter Hawkins 648a512488 [MHLO] Add direct MHLO lowerings for sparse primitives. 
PiperOrigin-RevId: 440374054
2022-04-08 08:43:57 -07:00
..
mlir
Integrate LLVM at llvm/llvm-project@eb27da7dec
2022-03-03 08:24:39 -08:00
BUILD
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
cpu_feature_guard.c
move #endif so that Windows doesn't have GetXCR0EAX defined twice
2021-06-04 17:26:27 -07:00
cpu_kernels.cc
schur lapack wrapper
2021-09-29 14:29:52 +02:00
cublas_kernels.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cublas_kernels.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cublas.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_gpu_kernel_helpers.cc
Fix return value for MakeBatchPointers.
2021-08-09 21:44:46 -07:00
cuda_gpu_kernel_helpers.h
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
cuda_linalg.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_linalg.py
[MHLO] Add direct MHLO lowerings for most linear algebra kernels.
2022-04-06 13:59:09 -07:00
cuda_lu_pivot_kernels.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_lu_pivot_kernels.cu.cc
Move all Abseil dependencies out of jaxlib CUDA libraries
2021-08-19 16:55:37 -07:00
cuda_lu_pivot_kernels.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_prng_kernels.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_prng_kernels.cu.cc
Move all Abseil dependencies out of jaxlib CUDA libraries
2021-08-19 16:55:37 -07:00
cuda_prng_kernels.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_prng.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
cuda_prng.py
[MHLO] Add MHLO lowering for PRNG kernels.
2022-04-06 13:23:01 -07:00
cusolver_kernels.cc
[JAX:GPU] Implement the full_matrices=False case of SVD without generating the full matrices and then slicing.
2022-03-04 05:55:36 -08:00
cusolver_kernels.h
[JAX:GPU] Implement the full_matrices=False case of SVD without generating the full matrices and then slicing.
2022-03-04 05:55:36 -08:00
cusolver.cc
[JAX:GPU] Implement the full_matrices=False case of SVD without generating the full matrices and then slicing.
2022-03-04 05:55:36 -08:00
cusolver.py
[MHLO] Add direct MHLO lowerings for most linear algebra kernels.
2022-04-06 13:59:09 -07:00
cusparse_kernels.cc
Fix typo: JAX_CUSPARSE_11030 -> JAX_CUSPARSE_11300
2022-03-02 18:32:21 -08:00
cusparse_kernels.h
Fix typo: JAX_CUSPARSE_11030 -> JAX_CUSPARSE_11300
2022-03-02 18:32:21 -08:00
cusparse.cc
Fix typo: JAX_CUSPARSE_11030 -> JAX_CUSPARSE_11300
2022-03-02 18:32:21 -08:00
cusparse.py
[MHLO] Add direct MHLO lowerings for sparse primitives.
2022-04-08 08:43:57 -07:00
gpu_kernels.cc
Fix typo: JAX_CUSPARSE_11030 -> JAX_CUSPARSE_11300
2022-03-02 18:32:21 -08:00
handle_pool.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
hip_gpu_kernel_helpers.cc
Merge pull request #9584 from ROCmSoftwarePlatform:rocm_refactor_jaxlib
2022-03-03 11:11:02 -08:00
hip_gpu_kernel_helpers.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_linalg.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_linalg.py
[MHLO] Add direct MHLO lowerings for most linear algebra kernels.
2022-04-06 13:59:09 -07:00
hip_lu_pivot_kernels.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_lu_pivot_kernels.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_lu_pivot_kernels.hip.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_prng_kernels.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_prng_kernels.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_prng_kernels.hip.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_prng.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hip_prng.py
[MHLO] Add MHLO lowering for PRNG kernels.
2022-04-06 13:23:01 -07:00
hipblas_kernels.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipblas_kernels.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipblas.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsolver_kernels.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsolver_kernels.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsolver.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsolver.py
[MHLO] Add direct MHLO lowerings for most linear algebra kernels.
2022-04-06 13:59:09 -07:00
hipsparse_kernels.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsparse_kernels.h
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsparse.cc
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
hipsparse.py
[MHLO] Add direct MHLO lowerings for sparse primitives.
2022-04-08 08:43:57 -07:00
init.py
Fix build.py to work on Linux once again.
2020-11-19 13:14:34 -05:00
jax.bzl
Merge pull request #9584 from ROCmSoftwarePlatform:rocm_refactor_jaxlib
2022-03-03 11:11:02 -08:00
kernel_helpers.h
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
kernel_pybind11_helpers.h
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
lapack_kernels_using_lapack.cc
Make lapack symbols strong in lapack_kernels
2022-01-25 16:30:30 -08:00
lapack_kernels.cc
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
lapack_kernels.h
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
lapack.cc
Merge pull request #8033 from SaturdayGenfo:schur-lapack-wrapper
2021-09-29 09:37:49 -07:00
lapack.py
[MHLO] Add direct MHLO lowerings for most linear algebra kernels.
2022-04-06 13:59:09 -07:00
pocketfft_kernels.cc
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
pocketfft_kernels.h
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
pocketfft.cc
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
pocketfft.fbs
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
pocketfft.py
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
setup.cfg
Export LICENSE.txt file in jaxlib wheels.
2020-12-11 10:20:03 -05:00
setup.py
Drop support for NumPy 1.18.
2022-01-04 12:11:38 -08:00
version.py
Update version numbers after jax/jaxlib release.
2022-04-07 16:40:19 -04:00