diff --git a/python/sglang/srt/disaggregation/prefill.py b/python/sglang/srt/disaggregation/prefill.py index d9016d0f9..7ab978a79 100644 --- a/python/sglang/srt/disaggregation/prefill.py +++ b/python/sglang/srt/disaggregation/prefill.py @@ -490,6 +490,9 @@ class SchedulerDisaggregationPrefillMixin: if copy_done is not None: copy_done.synchronize() + if result.routed_experts_output is not None: + result.routed_experts_output.finalize() + result.routed_experts_output = None logprob_pt = 0 # Transfer kv for prefill completed requests and add it into disagg_prefill_inflight_queue