Skip to content

weight-cache: support post-load lifecycle and MXFP4 IPC - #38306

Open
UNIDY2002 wants to merge 2 commits into
sgl-project:mainfrom
UNIDY2002:weight-cache-post-load-mxfp4-ipc
Open

UNIDY2002 wants to merge 2 commits into
sgl-project:mainfrom
UNIDY2002:weight-cache-post-load-mxfp4-ipc

Conversation

@UNIDY2002

@UNIDY2002 UNIDY2002 commented Sep 7, 2026 •

Copy link
Copy Markdown
Contributor

Motivation

Part of #33522.

Some models require a post-load logic. This PR also added support for MXFP4, so models like K3 can be loaded by weight cache.

Modifications

  • Added _post_load_weights() to the IPC loading lifecycle.
  • Added compressed-tensors to IPC_QUANT_ALLOWLIST
  • Some K3-specific patches,

Accuracy Tests

K3 and DSV3 can pass e2e.

Speed Tests and Profiling

Checklist

Review and Merge Process

  1. Ping Merge Oncalls to start the process. See the PR Merge Process.
  2. Get approvals from CODEOWNERS and other reviewers.
  3. Trigger CI tests with comments or contact authorized users to do so.
    • Common commands include /tag-and-rerun-ci, /tag-run-ci-label, /rerun-failed-ci
  4. After green CI and required approvals, ask Merge Oncalls or people with Write permission to merge the PR.

CI States

Latest PR Test (Base): 🚫 Run #34124802751
Latest PR Test (Extra): ✅ Run #34124802402
Latest PR Test (AMD ROCm 7.2): ❌ Run #34124802709

@UNIDY2002
UNIDY2002 force-pushed the weight-cache-post-load-mxfp4-ipc branch from a58ee20 to 1810aea Compare September 7, 2026 09:00
@UNIDY2002
UNIDY2002 force-pushed the weight-cache-post-load-mxfp4-ipc branch from 1810aea to 5d57b5c Compare September 7, 2026 09:06
@UNIDY2002

Copy link
Copy Markdown
Contributor Author

/tag-run-ci-label extra

@github-actions github-actions Bot added run-ci CI: run the baseline test suite on this PR run-ci-extra CI: also run the extra suite (requires run-ci) labels Sep 7, 2026
@UNIDY2002

Copy link
Copy Markdown
Contributor Author

/rerun-failed-ci

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

run-ci CI: run the baseline test suite on this PR run-ci-extra CI: also run the extra suite (requires run-ci)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant