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

Method __init__

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

Source from the content-addressed store, hash-verified

255 """The WGMMA instruction submodule."""
256
257 def __init__(self):
258 self.fence: Callable[..., Any] = _op_wrapper(_cuda_op.ptx_wgmma_fence)
259 self.commit_group = _op_wrapper(_cuda_op.ptx_wgmma_commit_group)
260 self.wait_group = _op_wrapper(_cuda_op.ptx_wgmma_wait_group)
261 self.noop_barrier = _op_wrapper(_cuda_op.ptx_wgmma_noop_barrier)
262 self.mma_async = WgmmaMmaAsyncNamespace()
263 self.encode_matrix_descriptor = _op_wrapper(_cuda_op.ptx_wgmma_encode_matrix_descriptor)
264
265
266class WgmmaMmaAsyncNamespace:

Callers

nothing calls this directly

Calls 2

_op_wrapperFunction · 0.90

Tested by

no test coverage detected