mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Peter Hawkins a48752a578 [MHLO] Remove most XLA translation rules. 
Almost all XLA translation rules have MHLO equivalents at this point, and there are no code paths that use the XLA translation rules in preference to their MLIR equivalents.

PiperOrigin-RevId: 442547482
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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] Remove most XLA translation rules.
2022-04-18 08:28:35 -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
[MHLO] Remove most XLA translation rules.
2022-04-18 08:28:35 -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
Copybara import of the project:
2022-04-16 22:37:09 -07:00