jax.experimental.pallas.mosaic_gpu.copy_gmem_to_smem

jax.experimental.pallas.mosaic_gpu.copy_gmem_to_smem(src, dst, barrier)

非同步地將 GMEM 參考複製到 SMEM 參考。

另請參閱

jax.experimental.mosaic.gpu.barrier_arrive() jax.experimental.mosaic.gpu.barrier_wait()

參數：:

• src (_Ref)

• dst (_Ref)

• barrier (_Ref)

回傳類型：:

None