dummy run corner case (#29433)

This commit is contained in:
Xieyang Xu 2025-11-25 16:20:35 -08:00 committed by GitHub
parent d8819c88eb
commit 12866af748
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2789,7 +2789,7 @@ class GPUModelRunner(
# returns True. before returning early here we call
# dummy run to ensure coordinate_batch_across_dp
# is called into to avoid out of sync issues.
self._dummy_run(1)
self._dummy_run(self._get_num_input_tokens(1))
if not has_kv_transfer_group():
# Return empty ModelRunnerOutput if no work to do.
return EMPTY_MODEL_RUNNER_OUTPUT