[executorch][cuda] Optimize short-query INT6 matvec kernels

[Gasoonjia]

Stack from ghstack (oldest at bottom):

• [executorch][cuda] Optimize short-query INT5 matvec kernels #21458
• -> [executorch][cuda] Optimize short-query INT6 matvec kernels #21457
• [executorch][cuda] Optimize short-query INT4 matvec kernels #21456

Add compile-time M=1/2/3/4 short-query paths and share their safe implementation through a ROWS/USE_SUM/GS16 device template. These kernels were generated and tuned with KernelAgent. Dispatch uses the regular gs16 kernels for M=1/2 and the precomputed-activation-sum kernels for M=3/4. The M=3 non-sum entry remains explicit as a performance/correctness comparison because templating that entry increased register pressure.

A100 op-level results (N=6656, gs=16, three seeds):

• M1, generic reference -> gs16: 0.039416 -> 0.038211 ms (1.032x; no regression)
• M2, generic reference -> gs16: 0.055348 -> 0.032540 ms (1.701x)
• M3, optimized non-sum -> activation-sum: 0.040003 -> 0.039949 ms (1.001x; neutral incremental change)
• M4, generic reference -> activation-sum: 0.100204 -> 0.045694 ms (2.193x)

The M3 baseline is intentionally the optimized non-sum kernel rather than the generic reference; this isolates the activation-sum optimization. All candidates are bitwise-correct across the three seeds, and ptxas reports zero spills.

Differential Revision: D114032330

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21457

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 3 New Failures, 2 Pending

As of commit 41b30bf with merge base e47f373 ():

NEW FAILURES - The following jobs have failed:

• Lint / lintrunner (gh)
  >>> Lint for backends/cuda/runtime/shims/tests/CMakeLists.txt:
• pull / unittest / linux / linux-job (gh)
  RuntimeError: Command docker exec -t 6cbaff476c4bf1ed6b589f46f5fda4a5b09b88d5550d915cf984e83223a345b9 /exec failed with exit code 1
• pull / unittest-editable / linux / linux-job (gh)
  RuntimeError: Command docker exec -t f3185e63cd35ec982b5a6e56ef86ca1568c0cd90c186ecc3e234b224b1434aba /exec failed with exit code 1

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.