For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

mbarrier_test_wait

def mbarrier_test_wait[type: AnyType](shared_mem: Pointer[type, address_space=AddressSpace.SHARED], state: Int) -> Bool

Test if all threads have arrived at the memory barrier.

Non-blocking check to see if all participating threads have reached the barrier. Only supported on NVIDIA GPUs.

Parameters:

Args:

shared_mem (Pointer[type, address_space=AddressSpace.SHARED]): Pointer to the shared memory barrier.
state (Int): Expected state of the memory barrier.

Returns:

Bool: True if all threads have arrived, False otherwise.