[XPU] support plugin model (#6092)

2026-04-23 00:17:25 +08:00 · 2026-01-20 13:00:09 +08:00
parent 988e0bc338
commit 45ebb2efb4
5 changed files with 51 additions and 16 deletions
@@ -328,7 +328,7 @@ def xpu_post_process_normal(

    # 2. Update the input buffer of the model
    with paddle.framework._no_check_dy2st_diff():
-        if envs.ENABLE_V1_KVCACHE_SCHEDULER and not skip_save_output:
+        if envs.ENABLE_V1_KVCACHE_SCHEDULER:
            update_inputs_v1(
                model_output.stop_flags,
                model_output.not_need_stop,