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

Class Tcgen05MmaNamespace

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

The Tcgen05 MMA instruction submodule.

Source from the content-addressed store, hash-verified

327
328
329class Tcgen05MmaNamespace:
330 """The Tcgen05 MMA instruction submodule."""
331
332 def __init__(self):
333 self.block_scale = _op_wrapper(_cuda_op.ptx_tcgen05_mma_block_scale)
334 self.sp = Tcgen05MmaSpNamespace()
335
336 def __call__(self, *args, **kwds):
337 return _op_wrapper(_cuda_op.ptx_tcgen05_mma)(*args, **kwds)
338
339 # __call__ corresponds to ptx_tcgen05_mma
340 __tir_call_op_name__ = "ptx_tcgen05_mma"
341
342
343class Tcgen05MmaSpNamespace:

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…