triton.experimental.gluon.language.amd.rdna4.wmma

triton.experimental.gluon.language.amd.rdna4.wmma(a, b, acc, _semantic=None)

Computes matrix-multiplication of a * b + acc using AMD WMMA instruction.

Parameters:

• a (tensor) – The operand a to be multiplied.

• b (tensor) – The operand b to be multiplied.

• acc (tensor) – The accumulator tensor.