[Metax][CI] e2e ci tests enable cuda graph (#6401)

2026-04-23 00:17:25 +08:00 · 2026-02-09 16:25:23 +08:00
parent fd56d85346
commit 268276e287
4 changed files with 11 additions and 12 deletions
@@ -911,6 +911,7 @@ def rebuild_padding(
            seq_lens_decoder,
            seq_lens_encoder,
            batch_id_per_token_output,
+            cu_seqlens_q_output,
            first_token_out,
            enable_logprob,
        )