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

Method __init__

python/tvm/backend/cuda/script.py:302–318  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

300 """The Tcgen05 instruction submodule."""
301
302 def __init__(self):
303 self.alloc = _op_wrapper(_cuda_op.ptx_tcgen05_alloc)
304 self.dealloc = _op_wrapper(_cuda_op.ptx_tcgen05_dealloc)
305 self.relinquish_alloc_permit = _op_wrapper(_cuda_op.ptx_tcgen05_relinquish_alloc_permit)
306 self.encode_matrix_descriptor = _op_wrapper(_cuda_op.ptx_tcgen05_encode_matrix_descriptor)
307 self.encode_instr_descriptor = _op_wrapper(_cuda_op.ptx_tcgen05_encode_instr_descriptor)
308 self.encode_instr_descriptor_block_scaled = _op_wrapper(
309 _cuda_op.ptx_tcgen05_encode_instr_descriptor_block_scaled
310 )
311 self.ld = _op_wrapper(_cuda_op.ptx_tcgen05_ld)
312 self.st = _op_wrapper(_cuda_op.ptx_tcgen05_st)
313 self.cp = _op_wrapper(_cuda_op.ptx_tcgen05_cp)
314 self.shift = _op_wrapper(_cuda_op.ptx_tcgen05_shift)
315 self.commit = _op_wrapper(_cuda_op.ptx_tcgen05_commit)
316 self.wait = Tcgen05WaitNamespace()
317 self.mma = Tcgen05MmaNamespace()
318 self.fence = Tcgen05FenceNamespace()
319
320
321class Tcgen05FenceNamespace:

Callers

nothing calls this directly

Calls 4

_op_wrapperFunction · 0.90
Tcgen05MmaNamespaceClass · 0.85

Tested by

no test coverage detected