tvm_tirx.backend.cuda.lang.TMABar

class tvm_tirx.backend.cuda.lang.TMABar(*args, **kwargs)

Bases: MBarrier

Barrier signaled by TMA (mbarrier.arrive.expect_tx).

When tx_count is None, falls back to a remote mbarrier.arrive (matching MBarrier.arrive defaults).

arrive(stage, tx_count=None, remote=None, pred=None)