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

Class WgmmaMmaAsyncNamespace

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

The WGMMA MMAAsync instruction submodule.

Source from the content-addressed store, hash-verified

264
265
266class WgmmaMmaAsyncNamespace:
267 """The WGMMA MMAAsync instruction submodule."""
268
269 def __init__(self):
270 self.ss = _op_wrapper(_cuda_op.ptx_wgmma_mma_async_ss)
271 self.rs = _op_wrapper(_cuda_op.ptx_wgmma_mma_async_rs)
272
273
274class MbarrierNamespace:

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…