mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/rocm
History
Rohit Santhanam 080cf47002 [ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork.
2022-06-29 14:34:08 +00: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
Replace references to absl::string_view with std::string_view.
2022-05-24 14:21:32 -07: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
Replace references to absl::string_view with std::string_view.
2022-05-24 14:21:32 -07: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
Fix hipblas kernels for ROCm.
2022-05-19 14:47:21 +00: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
[ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork.
2022-06-29 14:34:08 +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