[rl][sgl] fix tensor mismatch after pause (#21514)

This commit is contained in:
Bi Xue
2026-03-27 23:02:30 +08:00
committed by GitHub
parent 279e7738c5
commit 30397e0a1e
2 changed files with 217 additions and 1 deletions
+5 -1
View File
@@ -3246,7 +3246,11 @@ class Scheduler(
self.last_batch.filter_batch(
chunked_req_to_exclude=list(chunked_req_to_exclude)
)
self.running_batch.merge_batch(self.last_batch)
if not self.last_batch.is_empty():
if self.running_batch.is_empty():
self.running_batch = self.last_batch
else:
self.running_batch.merge_batch(self.last_batch)
self.last_batch = None
self.cur_batch = None