triton.language.dot

triton.language.dot(input, other)

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.