tvm_tirx.tirx.script.ptx_legacy

Contents

tvm_tirx.tirx.script.ptx_legacy#

Public canonical TVMScript dialect namespace.

Functions#

mma(a_layout, b_layout, a_dtype, b_dtype, ...)

Construct legacy MMA with element offsets and an optional bit operator.

ldmatrix(num, dtype, local_ptr, ...[, ty, loc])

Load a matrix with explicit pointer and element-offset operands.