Skip to main content
Log in

Mojo struct

TMABarrier

@register_passable(trivial) struct TMABarrier

Fields

  • mbar (UnsafePointer[SIMD[int64, 1], address_space=AddressSpace(3)]):

Implemented traits

AnyType, CollectionElement, Copyable, ExplicitlyCopyable, Movable, UnknownDestructibility

Methods

__init__

__init__() -> Self

__init__(addr: UnsafePointer[SIMD[int64, 1], address_space=AddressSpace(3), alignment=8]) -> Self

init

init(self, num_threads: SIMD[int32, 1] = __init__[__mlir_type.!kgen.int_literal](1))

expect_bytes

expect_bytes(self, bytes: SIMD[int32, 1])

wait

wait(self, phase: SIMD[uint32, 1] = __init__[__mlir_type.!kgen.int_literal](0))

arrive_cluster

arrive_cluster(self, cta_id: SIMD[uint32, 1], count: SIMD[uint32, 1] = __init__[__mlir_type.!kgen.int_literal](1))

arrive

arrive(self) -> Int