The MMA instruction submodule.
| 106 | |
| 107 | |
| 108 | class MmaNamespace: |
| 109 | """The MMA instruction submodule.""" |
| 110 | |
| 111 | def __init__(self): |
| 112 | self.sp = _dtype_forward(_cuda_op.ptx_mma_sp) |
| 113 | # Apache-compatible variant of ptx_mma. Coexists with the |
| 114 | # fork-native ``__call__`` form (``T.ptx.mma(...)``). |
| 115 | self.legacy = _dtype_forward(_cuda_op.ptx_mma_legacy) |
| 116 | # __call__ corresponds to ptx_mma |
| 117 | self.__tir_call_op_name__ = "ptx_mma" |
| 118 | |
| 119 | def __call__(self, *args, **kwds): |
| 120 | return _dtype_forward(_cuda_op.ptx_mma)(*args, **kwds) |
| 121 | |
| 122 | |
| 123 | class CpAsyncNamespace: |
no outgoing calls
no test coverage detected
searching dependent graphs…