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