Does gemm-streamk has stricter alignment requirement than normal gemm? #722
-
|
I find that problem sizes like (1, 512, 1024) (row, col -> row) will fail when using gemm-streamk, but when using normal gemm, this will not happen. So does gemm-streamk has stricter alignment requirement than normal gemm? |
Beta Was this translation helpful? Give feedback.
Answered by
jackkosaian
Dec 6, 2022
Replies: 1 comment 1 reply
-
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Thanks! This PR solved the problem exactly I have. |
Beta Was this translation helpful? Give feedback.
Answer selected by
lygztq
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi, @lygztq. Does this PR help?