triton.language.dot

triton.language.dot(input, other, trans_a=False, trans_b=False, allow_tf32=True)

Returns the matrix product of two blocks.

The two blocks must be two dimensionals and have compatible inner dimensions.

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

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