[CPU] Fix model failures on Xeon (#29497)

2026-07-02 13:20:18 +08:00
parent aff44d748d
commit cb06c4e6ce
6 changed files with 86 additions and 98 deletions
@@ -130,6 +130,9 @@ class EagerRunner(BaseRunner):
                if is_encoder_decoder
                else 0
            ),
+            encoder_lens_dtype=(
+                torch.int64 if torch.device(mr.device).type == "cpu" else torch.int32
+            ),
            dp_size=sa.dp_size,
        )
        # Eager has no capture step, so warm up here (run-once via mr._kernel_warmed_up).