mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 06:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Peter Hawkins 94203bf022
Update XLA. (#1837) 
Update jaxlib BUILD for ead06270dc
2019-12-10 11:25:09 -05:00
..
BUILD
Update XLA. (#1837)
2019-12-10 11:25:09 -05:00
cublas.cc
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08: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
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng_kernels.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng.py
Add a handwritten ThreeFry2x32 CUDA kernel. (#1756)
2019-11-24 13:06:23 -05:00
cusolver.cc
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cusolver.py
Implement batched singular value decomposition.
2019-09-05 18:12:00 -04:00
gpu_kernel_helpers.cc
Add error checking to PRNG CUDA kernel. (#1760)
2019-11-25 11:48:45 -05:00
gpu_kernel_helpers.h
Add error checking to PRNG CUDA kernel. (#1760)
2019-11-25 11:48:45 -05:00
kernel_helpers.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
kernel_pybind11_helpers.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
lapack.pyx
Fix integer overflow for large matrices in linear algebra kernels. (#1648)
2019-11-08 12:49:07 -08:00
pytree.cc
Avoid out-of-bounds dereference for arity-0 nodes. (#1713)
2019-11-18 15:35:07 -05:00
version.py
Bump jaxlib version to 0.1.37 and update WORKSPACE.
2019-12-03 12:29:34 -08:00