Skip to main content
Log in

Mojo function

mbarrier_arrive_expect_tx_shared

mbarrier_arrive_expect_tx_shared[type: AnyType](addr: UnsafePointer[type, address_space=AddressSpace(3), alignment=alignment, mut=mut, origin=origin], tx_count: SIMD[int32, 1])

Configure a shared memory barrier to expect additional async transactions.

Updates the current phase of the memory barrier to track completion of additional asynchronous transactions. Only supported on NVIDIA GPUs.

Args:

  • addr (UnsafePointer[type, address_space=AddressSpace(3), alignment=alignment, mut=mut, origin=origin]): Pointer to the shared memory barrier.
  • tx_count (SIMD[int32, 1]): Number of expected transactions to track.