jax.experimental.pallas.mosaic_gpu.barrier_wait

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

等待给定的屏障。

参数:

barrier (state.AbstractRef)

返回类型:

无