[executorch][cuda] Optimize short-query INT4 matvec kernels - #21473
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21473
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 Unclassified FailureAs of commit 2058ca5 with merge base f4a21b8 ( UNCLASSIFIED FAILURE - DrCI could not classify the following job because the workflow did not run on the merge base. The failure may be pre-existing on trunk or introduced by this PR:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
digantdesai
left a comment
There was a problem hiding this comment.
Review automatically exported from Phabricator review in Meta.
97c14de
into
gh/gasoonjia/178/base
This PR was created by the merge bot to help merge the original PR into the main branch. ghstack PR number: #21473 by @Gasoonjia ^ Please use this as the source of truth for the PR details, comments, and reviews ghstack PR base: https://github.com/pytorch/executorch/tree/gh/gasoonjia/178/base ghstack PR head: https://github.com/pytorch/executorch/tree/gh/gasoonjia/178/head Merge bot PR base: https://github.com/pytorch/executorch/tree/main Merge bot PR head: https://github.com/pytorch/executorch/tree/gh/gasoonjia/178/orig Differential Revision: [D114032326](https://our.internmc.facebook.com/intern/diff/D114032326/) @diff-train-skip-merge Co-authored-by: gasoonjia <gasoonjia@icloud.com>
Stack from ghstack (oldest at bottom):
Add compile-time M=2/3/4 fused short-query kernels while preserving the existing M=1 path. The shared ROWS template was reviewed and benchmarked by KernelAgent; KA found no faster safe rewrite, so the validated template is retained.
A100 kernel microbenchmarks versus main:
Correctness tests cover M=1,2,3,4 with three randomized seeds. ptxas reports zero spills.
Differential Revision: D114032326