Unverified Commit 6167c0e5 authored by 张宇's avatar 张宇 Committed by GitHub
Browse files

[Bugfix][Core] add seq_id_to_seq_group clearing to avoid memory leak when s… (#16472)


Signed-off-by: default avatar开哲 <kaizhe.zy@alibaba-inc.com>
Co-authored-by: default avatar开哲 <kaizhe.zy@alibaba-inc.com>
parent ed2e4646
...@@ -179,6 +179,13 @@ class RequestOutput: ...@@ -179,6 +179,13 @@ class RequestOutput:
group.finish_seq(seq_group) group.finish_seq(seq_group)
if assembled_seq_group is None: if assembled_seq_group is None:
return None return None
# clear finished seq in seq_id_to_seq_group
if len(group.to_be_finished) == 0:
for sub_request_id in list(group.seq_id_to_index.keys()):
if sub_request_id in seq_id_to_seq_group:
del seq_id_to_seq_group[sub_request_id]
return cls.from_seq_group(assembled_seq_group, use_cache, return cls.from_seq_group(assembled_seq_group, use_cache,
seq_id_to_seq_group) seq_id_to_seq_group)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment