jax.experimental.pallas.mosaic_gpu.commit_smem

目錄

jax.experimental.pallas.mosaic_gpu.commit_smem#

jax.experimental.pallas.mosaic_gpu.commit_smem()[原始碼]#: 提交所有寫入至 SMEM 的操作，使其對 loads、TMA 和 WGMMA 可見。