Skip to content

[Diffusion] Revert CPU AMX optimizations - #30716

Merged
BBuf merged 1 commit into
sgl-project:mainfrom
mickqian:codex/revert-diffusion-amx
Jul 10, 2026
Merged

[Diffusion] Revert CPU AMX optimizations#30716
BBuf merged 1 commit into
sgl-project:mainfrom
mickqian:codex/revert-diffusion-amx

Conversation

@mickqian

@mickqian mickqian commented Jul 10, 2026

Copy link
Copy Markdown
Collaborator

Reverts #28527 (177c048c68).

That change adds a second unconditional process_weights_after_loading() pass in the diffusion FSDP loader. The hook is not idempotent for quantized weights, and the added device_loading_context also breaks FSDP inference by attempting to restore CUDA-backed parameters to CPU storage.

CI evidence: multimodal-gen-test-2-gpu (2), where fsdp-inference fails in device_loading_context with a CPU/CUDA storage mismatch.


CI States

Latest PR Test (Base): ⏳ Run #29061437164
Latest PR Test (Extra): ❌ Run #29061437106

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@github-actions github-actions Bot added Multi-modal multi-modal language model sgl-kernel diffusion SGLang Diffusion labels Jul 10, 2026
@mickqian
mickqian marked this pull request as ready for review July 10, 2026 01:03
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@mickqian

Copy link
Copy Markdown
Collaborator Author

@mingfeima, please make sure all nvidia multimodal-gen tests (call-multimodal-gen-tests) pass before merging. Join me in making the CI more stable and pushing SGLang-Diffusion forward.

@mickqian

Copy link
Copy Markdown
Collaborator Author

/tag-and-rerun-ci

@BBuf
BBuf merged commit 5ce5e1e into sgl-project:main Jul 10, 2026
133 of 186 checks passed
Chronostasys pushed a commit to MindLab-Research/sglang that referenced this pull request Aug 24, 2026
jakki-amd pushed a commit to jakki-amd/sglang that referenced this pull request Sep 9, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

diffusion SGLang Diffusion Multi-modal multi-modal language model run-ci sgl-kernel

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants