mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/rocm
History
Peter Hawkins bb0816227d Add a batched QR decomposition implementation on GPU. 
PiperOrigin-RevId: 449583027
2022-05-18 14:50:18 -07:00
..
BUILD.bazel
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
hip_gpu_kernel_helpers.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_gpu_kernel_helpers.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_linalg.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_lu_pivot_kernels.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_lu_pivot_kernels.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_lu_pivot_kernels.hip.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_prng_kernels.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_prng_kernels.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_prng_kernels.hip.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hip_prng.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hipblas_kernels.cc
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
hipblas_kernels.h
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
hipblas.cc
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
hipsolver_kernels.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hipsolver_kernels.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipsolver.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hipsparse_kernels.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hipsparse_kernels.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipsparse.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00