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 08c3c2ec24 Split CUDA and HIP C++ code in jaxlib into separate directories. 
PiperOrigin-RevId: 447062506
2022-05-06 13:48:00 -07:00
..
BUILD.bazel
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_gpu_kernel_helpers.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07: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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_lu_pivot_kernels.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_prng_kernels.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hip_prng.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipblas_kernels.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07: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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipsolver_kernels.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07: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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00
hipsparse_kernels.cc
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07: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
Split CUDA and HIP C++ code in jaxlib into separate directories.
2022-05-06 13:48:00 -07:00