jax.experimental.pallas.mosaic_gpu.commit_smem

目录

jax.experimental.pallas.mosaic_gpu.commit_smem#

jax.experimental.pallas.mosaic_gpu.commit_smem()[源代码]#: 提交所有对 SMEM 的写入，使其对加载、TMA 和 WGMMA 可见。