What are the differences between segment_matmul and grouped_matmul ?
#206
Answered
by
rusty1s
clearhanhui
asked this question in
Q&A
-
|
The cpp source code contains segment_matmul and grouped_matmul. It looks like they function the same, just with different input arguments? |
Beta Was this translation helpful? Give feedback.
Answered by
rusty1s
Feb 27, 2023
Replies: 1 comment
-
|
Yes, they are using the same underlying algorithm, but one takes list of tensors as input and one expects a contiguous memory layout. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
clearhanhui
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Yes, they are using the same underlying algorithm, but one takes list of tensors as input and one expects a contiguous memory layout.