jax.experimental.pallas.mosaic_gpu.barrier_wait

jax.experimental.pallas.mosaic_gpu.barrier_wait(barrier)

等待给定的屏障。

参数:

barrier (pallas_core.AbstractMemoryRef)

返回类型:

None