Skip to content

Fix MDP partitioning for scoped ONNX aliases - #1430

Merged
asmigosw merged 5 commits into
qualcomm:mainfrom
asmigosw:mdp_partition
Oct 13, 2026
Merged

asmigosw merged 5 commits into
qualcomm:mainfrom
asmigosw:mdp_partition

Conversation

@asmigosw

Copy link
Copy Markdown
Contributor

Fix PP2 prefill compilation failures caused by scoped node-name collisions such as /Gather_5 and /language_model/Gather_5. The MDP generator now aligns alias partitions with the QAIC compiler dump and preserves compiler ordering.

Verified both ONNX and intersection MDP strategies compile successfully.

Comment thread QEfficient/compile/mdp_generator.py
Comment thread QEfficient/compile/mdp_generator.py
Comment thread QEfficient/compile/mdp_generator.py
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
)
)

mdp_json = generate_disagg_mdp_intersection_config(

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Please extend this regression to run precheck_and_autofix_mdp_partition_order() after generate_disagg_mdp_intersection_config() and assert /Gather_5 is still retained.

The production path runs this precheck after the scoped-alias repair, so the current test may pass even if the precheck later removes /Gather_5 before the final MDP JSON is written.

Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>

@quic-mamta Mamta Singh (quic-mamta) left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
@asmigosw
asmigosw merged commit 19c2fd4 into qualcomm:main Oct 13, 2026
19 checks passed
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.

2 participants