jax.experimental.pallas.mosaic_gpu.commit_smem

目录

jax.experimental.pallas.mosaic_gpu.commit_smem#

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