MCPcopy Create free account
hub / github.com/apache/tvm / MbarrierNamespace

Class MbarrierNamespace

python/tvm/backend/cuda/script.py:274–282  ·  view source on GitHub ↗

The Mbarrier instruction submodule.

Source from the content-addressed store, hash-verified

272
273
274class MbarrierNamespace:
275 """The Mbarrier instruction submodule."""
276
277 def __init__(self):
278 self.init = _op_wrapper(_cuda_op.ptx_mbarrier_init)
279 self.try_wait = _op_wrapper(_cuda_op.ptx_mbarrier_try_wait)
280 self.try_wait_once = _op_wrapper(_cuda_op.ptx_mbarrier_try_wait_once)
281 self.try_wait_acquire_cluster = _op_wrapper(_cuda_op.ptx_mbarrier_try_wait_acquire_cluster)
282 self.arrive = MbarrierArriveNamespace()
283
284
285class MbarrierArriveNamespace:

Callers 1

__init__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…