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

Method __init__

python/tvm/backend/cuda/script.py:161–165  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

159 """The CpAsyncBulk instruction submodule."""
160
161 def __init__(self):
162 self.commit_group = _op_wrapper(_cuda_op.ptx_cp_async_bulk_commit_group)
163 self.wait_group = _op_wrapper(_cuda_op.ptx_cp_async_bulk_wait_group)
164 self.tensor = CpAsyncBulkTensorNamespace()
165 self.s2c = _op_wrapper(_cuda_op.ptx_cp_async_bulk_shared_to_cluster)
166
167 def __call__(self, *args, **kwds):
168 return _dtype_forward(_cuda_op.ptx_cp_async_bulk)(*args, **kwds)

Callers

nothing calls this directly

Calls 2

_op_wrapperFunction · 0.90

Tested by

no test coverage detected