jax.experimental.pallas.mosaic_gpu.commit_smem

Contents

jax.experimental.pallas.mosaic_gpu.commit_smem#

jax.experimental.pallas.mosaic_gpu.commit_smem()[source]#: Commits all writes to SMEM, making them visible to loads, TMA and WGMMA.