onnxruntime
QMoE: prepack int4/int8 expert weights in PrePack hook (symmetric with MatMulNBits)
#28749
Merged

QMoE: prepack int4/int8 expert weights in PrePack hook (symmetric with MatMulNBits) #28749

justinchuby
justinchuby QMoE: prepack int4/int8 expert weights in PrePack hook
834eae86
justinchuby Add TestQMoEIntPrePackParity for the new QMoE.PrePack hook
66c7f47e
justinchuby Fix import order in test_qmoe_cuda.py
516812cb
justinchuby Rewrite QMoE PrePack test as a smoke test; verified passing on H200
197819fb
justinchuby
tianleiwu
tianleiwu requested changes on 2026-06-02
justinchuby QMoE: address review — opt-in weights_prepacked attribute, SM guard, …
2fcb9404
justinchuby
justinchuby QMoE: cache weight shapes during PrePack so source initializer can be…
5e1491c7
justinchuby
justinchuby justinchuby requested a review from tianleiwu tianleiwu 47 days ago
tianleiwu
tianleiwu requested changes on 2026-06-02
tianleiwu
tianleiwu commented on 2026-06-03
justinchuby Address round-3 QMoE int-prepack review feedback
360217b4
justinchuby justinchuby requested a review from tianleiwu tianleiwu 46 days ago
justinchuby justinchuby requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 46 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-06-03
justinchuby QMoE: align PrePack test name/docs with smoke-test intent
3cbbf512
justinchuby
tianleiwu
tianleiwu commented on 2026-06-03
tianleiwu
tianleiwu commented on 2026-06-03
justinchuby QMoE: pass SM75 through prepack clamp to mirror dispatch
3d4e0be1
justinchuby
justinchuby QMoE: make weights_prepacked an optional tri-state attribute
4d898bf1
justinchuby QMoE: set weights_prepacked schema default to -1 (auto)
543ab255
justinchuby justinchuby requested a review from tianleiwu tianleiwu 45 days ago
justinchuby justinchuby requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 45 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-06-04
tianleiwu
justinchuby
tianleiwu
tianleiwu dismissed these changes on 2026-06-05
justinchuby Merge remote-tracking branch 'origin/main' into qmoe-int-prepack
da0178c7
justinchuby QMoE: address round-4 int-prepack review feedback
e9cea0d3
justinchuby justinchuby dismissed their stale review via e9cea0d3 41 days ago
justinchuby justinchuby enabled auto-merge (squash) 40 days ago
justinchuby justinchuby requested a review from tianleiwu tianleiwu 40 days ago
justinchuby justinchuby requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 40 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-06-09
tianleiwu
tianleiwu dismissed these changes on 2026-06-09
tianleiwu tianleiwu dismissed their stale review 40 days ago
Dismissing: this review was posted with stale content from a different PR by mistake. Re-posting the correct review for PR #28749.
tianleiwu
tianleiwu commented on 2026-06-09
disabled auto-merge 40 days ago
Manually disabled by user
justinchuby justinchuby enabled auto-merge (squash) 40 days ago
tianleiwu
tianleiwu approved these changes on 2026-06-09
justinchuby justinchuby merged 1e1f574c into main 40 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone