[Iluvatar GPU] fix ci error caused by rebuild_padding param and cuda graph (#4504)

2025-12-24 13:28:13 +08:00 · 2025-10-21 21:41:41 +08:00
parent d70aacfbdc
commit dc7facaa7f
4 changed files with 95 additions and 36 deletions
--- a/fastdeploy/model_executor/pre_and_post_process.py
+++ b/fastdeploy/model_executor/pre_and_post_process.py
@@ -732,7 +732,9 @@ def rebuild_padding(
            seq_lens_decoder,
            seq_lens_encoder,
            output_padding_offset,
+            first_token_out,
            max_input_length,
+            enable_logprob,
        )
    elif current_platform.is_gcu():
        from fastdeploy.model_executor.ops.gcu import rebuild_padding