[NPU][Bugfix] Accept in_capture in Ascend replay metadata (#29598)

2026-06-29 14:37:45 +08:00
parent cd91dd0757
commit 62d7929b3a
1 changed files with 1 additions and 0 deletions
@@ -131,6 +131,7 @@ class AscendMambaAttnBackendBase(MambaAttnBackendBase):
        spec_info: Optional[SpecInput],
        seq_lens_cpu: Optional[torch.Tensor],
        num_padding: Optional[int] = None,
+        in_capture: bool = False,
    ):
        # out_graph passes seq_lens_cpu=None at capture; mirror the base guard.
        if seq_lens_cpu is None: