Confused about “Matrix Multiplication“ in tutorial #5379
Unanswered
HonestDeng
asked this question in
Q&A
Replies: 1 comment
-
Ok. I find out that "dot" of triton is not just "dot product" of 2 vectors or element-wise matrix multiplication. When the the inputs of dot is 2D, "dot" actually performs standard matrix mulplication. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
In the tutorial, the mattrix multiplication is described as:
In the second last line code, sub-matrix a is BLOCK_SIZE_M by BLOCK_SIZE_K, and sub-matrix b is BLOCK_SIZE_K by BLOCK_SIZE_N. Their shapes is not compatible to perform "dot" application. Why?
Beta Was this translation helpful? Give feedback.
All reactions