mirror of
https://github.com/ROCm/jax.git
synced 2025-04-23 20:16:06 +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