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
-
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