mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00

This allows us to significantly simplify the generated PTX/SASS, which is currently cluttered with LLVM trying to align slices to start at bit 0 and failing to CSE the right shifts. PiperOrigin-RevId: 737967890