[AMD] Fix DeepSeek MLA prefill shape mismatch on HIP eager fallback (missing mha_companion_layers) (#31675)

2026-07-19 01:40:41 -07:00
parent 942bf04ef9
commit c68392c535
1 changed files with 1 additions and 0 deletions
@@ -317,6 +317,7 @@ class EagerRunner(BaseRunner):
                        model_runner.moe_layers,
                        model_runner.moe_fusions,
                        dsa_indexers=model_runner.dsa_indexers,
+                        mha_companion_layers=model_runner.mha_companion_layers,
                    ),
                ):
                    ret = model_runner.model.forward(