tvm_tirx.backend.cuda.script.wgmma

Contents

tvm_tirx.backend.cuda.script.wgmma#

CUDA TVMScript namespaces.

Functions#

noop_barrier(*[, attrs, ty_args, loc, ty])

Construct a call to tirx.cuda.wgmma_noop_barrier.

encode_matrix_descriptor(*[, attrs, ...])

Construct a call to tirx.cuda.wgmma_encode_matrix_descriptor.
Contents