triton.dot

triton.dot(input, other, builder=None)

Returns the matrix product of two blocks. The two blocks must be two dimensionals and have compatible inner dimensions.

Parameters

• input (2D block of scalar-type in {float16, float32}) – The first block to be multiplied.

• other (2D block of scalar-type in {float16, float32}) – The second block to be multiplied.

• builder (triton.ir.builder) – IR builder to generate code into, optional from within @triton.jit functions