Skip to content

[Draft] Add MatMulNBits estimation test matrices and profiling validation - #34211

Draft
Tommaso Adani (tadani3) wants to merge 36 commits into
mainfrom
tommasoadani/matmul_estimation_experiments
Draft

Tommaso Adani (tadani3) wants to merge 36 commits into
mainfrom
tommasoadani/matmul_estimation_experiments

Conversation

@tadani3

Copy link
Copy Markdown

Description

Add reproducible tooling and experimental evidence for validating
route-aware MatMulNBits estimates:

  • Deterministic single-operator cases and compute-mode/bias/tail matrices.
  • CPU runner checking capture completeness, logical counts, actual routes,
    and reporting-enabled/disabled output equivalence.
  • Linux perf collection with repeated trials and CSV/Markdown summaries.
  • CUDA single-case capture and NVTX profiling driver.
  • Testing strategy documenting experiment gates and deferred refinements.

Record CPU audits covering aligned shapes, bias/tails, partial K blocks,
explicit zero points, zero activation blocks, and matched FP16/FP32 inputs.
Compare hardware-measured recurring process FP work with independently
derived, bounded arithmetic models.

Production formula refinements remain deferred. CUDA execution/Nsight
qualification and broader CPU coverage are still pending.

Motivation and Context

Establish repeatable evidence for evaluating MatMulNBits estimates before
changing their formulas. Separate logical-count correctness, runtime capture
validation, and implementation-specific FP-work accuracy.

Document observed discrepancies, kernel-level explanations, held-out checks,
and explicit limitations rather than treating latency or passing estimator
tests as proof of executed-FLOP accuracy.

Tommaso Adani and others added 30 commits October 15, 2026 14:57
Forward the actual workload collector through eight GPT beam/greedy and T5/Whisper calls. Preserve the T5 encoder/decoder MatMulNBits regression sequences while covering collector propagation.

Validation: CPU Debug builds successful; 25 collector tests passed with 1 LUT skipped; 26 search regression tests passed.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Enable logical core and bias reporting for successful CUDA submissions, explicitly exclude graph capture/replay, and add CPU/CUDA parity regressions. Fix the CI warning test's inherited logging severity.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
…nt core work, bias additions, and nominal dequantization/scaling by selected CPU route. Preserve existing logical counts in a separate reporting section.
…A routes through append-only provider transport and estimate core work, bias additions, and nominal dequantization.
…and add CPU estimator validation, an 84-case matrix, and Linux perf collection with Markdown/CSV reporting, documenting measurement scope and accuracy limitations.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant