jax.experimental.pallas.mosaic_gpu.Barrier# class jax.experimental.pallas.mosaic_gpu.Barrier(num_arrivals: 'int', num_barriers: 'int' = 1)[source]# Parameters: num_arrivals (int) num_barriers (int) __init__(num_arrivals, num_barriers=1)# Parameters: num_arrivals (int) num_barriers (int) Return type: None Methods __init__(num_arrivals[, num_barriers]) get_ref_aval() Attributes num_barriers num_arrivals
