jax.experimental.pallas.mosaic_gpu.wait_smem_to_gmem# jax.experimental.pallas.mosaic_gpu.wait_smem_to_gmem(n, wait_read_only=False)[源代码]# 等待飞行中的 SMEM->GMEM 复制的数量不超过 n 个。 参数: n (int) – 等待的最大飞行中复制数量。 wait_read_only (bool) – 如果为 True，则等待飞行中的复制完成从 SMEM 读取。不等待写入 GMEM。 返回类型: 无
