mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Jake VanderPlas f8081a9a52 [sparse] fix GPU translation rule for coo/csr matmat
2021-08-10 10:13:29 -07:00
..
BUILD
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -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
cublas.cc
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -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.py
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
cuda_lu_pivot_kernels.cc
Reimplement as JAX Primitive
2021-04-03 14:11:36 -07:00
cuda_lu_pivot_kernels.cu.cc
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
cuda_lu_pivot_kernels.h
In OSS #include "third_party/tensorflow/..." should be #include "tensorflow/..."
2021-08-09 20:46:10 -07: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
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
cuda_prng_kernels.h
In OSS #include "third_party/tensorflow/..." should be #include "tensorflow/..."
2021-08-09 20:46:10 -07:00
cuda_prng.py
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
cusolver.cc
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
cusolver.py
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
cusparse.cc
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
cusparse.py
[sparse] fix GPU translation rule for coo/csr matmat
2021-08-10 10:13:29 -07:00
handle_pool.h
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
init.py
Fix build.py to work on Linux once again.
2020-11-19 13:14:34 -05:00
jax.bzl
Remove jax/BUILD file.
2021-05-19 12:55:34 -07: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.pyx
Initialize variables in LAPACK work size queries to prevent false positives in memory sanitizers.
2021-02-01 16:38:24 -08: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
Update flatbuffer ver2 EndVector usage.
2021-05-10 16:45:19 -07:00
rocblas.cc
Use absl::Status::message() instead of error_message().
2021-08-09 23:44:36 -07:00
rocm_gpu_kernel_helpers.cc
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
rocm_gpu_kernel_helpers.h
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
rocsolver.py
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
setup.cfg
Export LICENSE.txt file in jaxlib wheels.
2020-12-11 10:20:03 -05:00
setup.py
Type stubs for jaxlib.xla_extension no longer use -stubs suffix
2021-08-02 14:31:11 +01:00
version.py
Upgrade versions after jaxlib release
2021-08-09 16:37:44 -07:00