jax.experimental.pallas.mosaic_gpu.barrier_arrive#

jax.experimental.pallas.mosaic_gpu.barrier_arrive(barrier)[source]#

Arrives at the given barrier.

Parameters:

barrier (pallas_core.AbstractMemoryRef)

Return type:

None