jax.experimental.pallas.mosaic_gpu.barrier_wait

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

Waits on the given barrier.

Parameters:

barrier (pallas_core.AbstractMemoryRef)

Return type:

None