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 aa7d291767 Replace references to absl::string_view with std::string_view. 
PiperOrigin-RevId: 450768333
2022-05-24 14:21:32 -07:00
..
cuda
Replace references to absl::string_view with std::string_view.
2022-05-24 14:21:32 -07:00
mlir
Include mlir.transforms and mlir.passmanager in jaxlib BUILD.
2022-05-16 13:00:22 +00:00
rocm
Replace references to absl::string_view with std::string_view.
2022-05-24 14:21:32 -07:00
BUILD
Remove unused jaxlib build targets.
2022-05-06 15:34: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
gpu_linalg.py
[ROCm] Fix hip_lu_pivots_to_permutation.
2022-05-23 13:58:59 +00:00
gpu_prng.py
Fix unit test failures on ROCm.
2022-05-09 13:36:48 +00:00
gpu_solver.py
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
gpu_sparse.py
Refactor custom call building code in jaxlib to use a helper function.
2022-05-06 14:51:24 -07:00
handle_pool.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
init.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07: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
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
mhlo_helpers.py
Refactor custom call building code in jaxlib to use a helper function.
2022-05-06 14:51:24 -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
Refactor custom call building code in jaxlib to use a helper function.
2022-05-06 14:51:24 -07:00
setup.cfg
Export LICENSE.txt file in jaxlib wheels.
2020-12-11 10:20:03 -05:00
setup.py
Include mlir.transforms and mlir.passmanager in jaxlib BUILD.
2022-05-16 13:00:22 +00:00
version.py
Updates values after jax and jaxlib 0.3.10 release
2022-05-04 21:17:37 -07:00