jax.experimental.pallas.mosaic_gpu.barrier_arrive#

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

到达给定的屏障。

参数:

barrier (state.AbstractRef)

返回类型: