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 bbdcec84f8 Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
..
BUILD.bazel
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00: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
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00:00
hipblas_kernels.h
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipblas.cc
Fixes to enable JAX to build on ROCm.
2022-05-06 22:57:51 +00: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