The CpAsyncBulk instruction submodule.
| 156 | |
| 157 | |
| 158 | class CpAsyncBulkNamespace: |
| 159 | """The CpAsyncBulk instruction submodule.""" |
| 160 | |
| 161 | def __init__(self): |
| 162 | self.commit_group = _op_wrapper(_cuda_op.ptx_cp_async_bulk_commit_group) |
| 163 | self.wait_group = _op_wrapper(_cuda_op.ptx_cp_async_bulk_wait_group) |
| 164 | self.tensor = CpAsyncBulkTensorNamespace() |
| 165 | self.s2c = _op_wrapper(_cuda_op.ptx_cp_async_bulk_shared_to_cluster) |
| 166 | |
| 167 | def __call__(self, *args, **kwds): |
| 168 | return _dtype_forward(_cuda_op.ptx_cp_async_bulk)(*args, **kwds) |
| 169 | |
| 170 | # __call__ corresponds to ptx_cp_async_bulk |
| 171 | __tir_call_op_name__ = "ptx_cp_async_bulk" |
| 172 | |
| 173 | |
| 174 | class CpAsyncBulkTensorNamespace: |
no outgoing calls
no test coverage detected
searching dependent graphs…