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

The annotation on async_load didn't indicate its write to SMEM, allowing it to be DCEd by MLIR canonicalization. We don't get much mileage out of those annotations, so let's just delete them for simplicity. PiperOrigin-RevId: 731003033