[mgpu] Add missing `commit_smem`s in `mosaic_gpu_test`.
PiperOrigin-RevId: 971120011
C
Chris Jones committed
950509f89cbab3022a288a2ad59656e6dc8222bd
Parent: 764425f
Committed by jax authors <google-ml-automation@google.com>
on 8/26/2026, 9:02:28 AM