You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
the file is include/cutlass/conv/device/implicit_gemm_convolution.h
in implicit gemm conv, why does not support split-k when group !=1 ?
is this for performance ?
The text was updated successfully, but these errors were encountered:
preFiredman
changed the title
[QST] implicit gemm conv does not support split-k currently ?
[QST] typo
Jan 21, 2025
preFiredman
changed the title
[QST] typo
[QST] why does not support split-k when group !=1 ?
Jan 22, 2025
preFiredman
changed the title
[QST] why does not support split-k when group !=1 ?
[QST] in implicit gemm conv, why does not support split-k when group !=1 ?
Jan 22, 2025
For group fprop/dgrad, usually C/K per group is small, e.g. < 64, and RS dimension of filter is not large, e.g. 3x3, so there is no need to do split k. In other words, group is already used to split the C/K dimension in fprop/dgrad, there is no need to use split_k to split it again.
the file is include/cutlass/conv/device/implicit_gemm_convolution.h
in implicit gemm conv, why does not support split-k when group !=1 ?
is this for performance ?
The text was updated successfully, but these errors were encountered: