jax.experimental.pallas.mosaic_gpu.barrier_wait#

jax.experimental.pallas.mosaic_gpu.barrier_wait(barrier)[原始碼]#

等待給定的 barrier。

參數:

barrier (pallas_core.AbstractMemoryRef)

返回類型:

None