jax.experimental.pallas.mosaic_gpu.barrier_arrive#

jax.experimental.pallas.mosaic_gpu.barrier_arrive(barrier)[源代码]#

到达给定的屏障。

参数:

barrier (pallas_core.AbstractMemoryRef)

返回类型:

None