jax.experimental.pallas.mosaic_gpu.commit_smem

目录

jax.experimental.pallas.mosaic_gpu.commit_smem#

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