Skip to content

feat: support flux model on mlu device.#1138

Open
phantomlei3 wants to merge 7 commits intojd-opensource:mainfrom
phantomlei3:feat/support-flux
Open

feat: support flux model on mlu device.#1138
phantomlei3 wants to merge 7 commits intojd-opensource:mainfrom
phantomlei3:feat/support-flux

Conversation

@phantomlei3
Copy link
Copy Markdown
Collaborator

No description provided.

@XuZhang99 XuZhang99 changed the title feat: support flux model on MLU platform. feat: support flux model on MLU device. Mar 30, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request extends support for CUDA and MLU backends within the DiT model framework by updating preprocessor guards and including necessary headers for Flux pipelines. It also modifies flag validation for the MLU backend to accommodate the 'dit' backend. A critical issue was identified in xllm/xllm.cpp where the model_type variable is shadowed inside an if block, causing it to remain empty in the outer scope and potentially leading to incorrect validation logic.

@XuZhang99 XuZhang99 changed the title feat: support flux model on MLU device. feat: support flux model on mlu device. Mar 30, 2026
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
XuZhang99
XuZhang99 previously approved these changes Mar 30, 2026
XuZhang99
XuZhang99 previously approved these changes Apr 2, 2026
yq33victor
yq33victor previously approved these changes Apr 2, 2026
Copy link
Copy Markdown
Collaborator

@yq33victor yq33victor left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@phantomlei3 phantomlei3 dismissed stale reviews from yq33victor and XuZhang99 via 347eee5 April 2, 2026 23:43
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.

4 participants