The WGMMA instruction submodule.
| 252 | |
| 253 | |
| 254 | class WgmmaNamespace: |
| 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 | |
| 266 | class WgmmaMmaAsyncNamespace: |
no outgoing calls
no test coverage detected
searching dependent graphs…