[Mosaic GPU] Reland back skipping multimem address initialization.

multimem_addr call is now skipped for all modules which doesn't use any mulimem operations.

Reverts af08a3be01b58cbc1acea0005165933b8e8fa882

PiperOrigin-RevId: 965196260

Levon Ter-Grigoryan committed 19d ago

1b23d21ca658731fdd33dc0f203b3f8ce93e2fc2

Parent: 38f2e8d

Committed by jax authors <google-ml-automation@google.com> on 8/15/2026, 2:28:35 PM