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