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
Peter Hawkins a852710a09 Merge CUDA and ROCM kernel code in jaxlib. 
The code for both CUDA and ROCM is almost identical, so with a small shim library to handle the differences we can share almost everything.

PiperOrigin-RevId: 483666051
2022-10-25 07:23:34 -07:00
..
BUILD.bazel
Merge CUDA and ROCM kernel code in jaxlib.
2022-10-25 07:23:34 -07:00