triton.language.dot¶
- triton.language.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, optional from within JIT'ed functions) – IR builder to generate code into