jax.experimental.pallas.mosaic_gpu.Barrier#

class jax.experimental.pallas.mosaic_gpu.Barrier(num_arrivals: 'int', num_barriers: 'int' = 1)[源代码]#
参数:

  • num_arrivals (int)

  • num_barriers (int)

__init__(num_arrivals, num_barriers=1)#
参数:

  • num_arrivals (int)

  • num_barriers (int)

返回类型:

None

方法

__init__(num_arrivals[, num_barriers])

get_ref_aval()

属性

num_barriers

num_arrivals