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

Class MbarrierArriveNamespace

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

The Mbarrier Arrive instruction submodule.

Source from the content-addressed store, hash-verified

283
284
285class MbarrierArriveNamespace:
286 """The Mbarrier Arrive instruction submodule."""
287
288 def __init__(self):
289 self.expect_tx = _op_wrapper(_cuda_op.ptx_mbarrier_arrive_expect_tx)
290 self.cluster_count = _op_wrapper(_cuda_op.ptx_mbarrier_arrive_cluster_count)
291
292 def __call__(self, *args, **kwds):
293 return _op_wrapper(_cuda_op.ptx_mbarrier_arrive)(*args, **kwds)
294
295 # __call__ corresponds to ptx_mbarrier_arrive
296 __tir_call_op_name__ = "ptx_mbarrier_arrive"
297
298
299class Tcgen05Namespace:

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…