Rohit Santhanam 080cf47002 [ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork.

2022-06-29 14:34:08 +00:00

..

cuda

Replace references to absl::string_view with std::string_view.

2022-05-24 14:21:32 -07:00

mlir

Rename the file to .cc to fix the broken builds

2022-06-09 09:18:07 -07:00

rocm

[ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork.

2022-06-29 14:34:08 +00: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

[mhlo] ConstOp -> ConstantOp, a la HLO

2022-06-14 16:33:24 -07: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

[mhlo] ConstOp -> ConstantOp, a la HLO

2022-06-14 16:33:24 -07:00

setup.cfg

Export LICENSE.txt file in jaxlib wheels.

2020-12-11 10:20:03 -05:00

setup.py

Increase the minimum scipy version to 1.5.

2022-06-24 15:07:09 -04:00

version.py

Update versions for jax/jaxlib release

2022-06-21 12:57:28 -07:00