jax

eb2fe971 - [Mosaic GPU] Add an additional WG barrier before copy_gmem_to_smem

[Mosaic GPU] Add an additional WG barrier before copy_gmem_to_smem This is necessary to ensure that all SMEM reads issued from a current WG have completed before we schedule the copy (that acts as an SMEM write)! PiperOrigin-RevId: 757647993