fix: populate batch req rids and per-request http_worker_ipc for mult… (#29882)
This commit is contained in:
@@ -1,5 +1,7 @@
|
||||
import unittest
|
||||
|
||||
import requests
|
||||
|
||||
from sglang.srt.utils import kill_process_tree
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
from sglang.test.kits.eval_accuracy_kit import MMLUMixin
|
||||
@@ -17,8 +19,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=211, stage="base-b", runner_config="1-gpu-large")
|
||||
register_amd_ci(est_time=345, suite="stage-b-test-1-gpu-small-amd")
|
||||
register_cuda_ci(est_time=220, stage="base-b", runner_config="1-gpu-large")
|
||||
register_amd_ci(est_time=355, suite="stage-b-test-1-gpu-small-amd")
|
||||
|
||||
|
||||
class TestMultiTokenizer(CustomTestCase, MMLUMixin):
|
||||
@@ -75,6 +77,40 @@ class TestMultiTokenizer(CustomTestCase, MMLUMixin):
|
||||
self.assertLess(res["median_ttft_ms"], 130 if is_in_amd_ci() else 86)
|
||||
self.assertLess(res["median_itl_ms"], 10)
|
||||
|
||||
def test_batch_input_ids_routing(self):
|
||||
# Regression guard for sgl-project/sglang#29878 (introduced by #29214).
|
||||
#
|
||||
# A batch of pre-tokenized `input_ids` (no text / multimodal) is the one
|
||||
# case that takes the batch-tokenization path (_send_batch_request ->
|
||||
# BatchTokenizedGenerateReqInput). In multi-tokenizer mode this batch
|
||||
# must stamp each sub-request's `http_worker_ipc` so the scheduler can
|
||||
# route every reply back to its owning tokenizer worker. If it is missing,
|
||||
# the requests hang forever.
|
||||
#
|
||||
# The existing ttft test only sends *text*, so it never exercises this
|
||||
# path — this case does, and uses a short timeout so a routing hang
|
||||
# fails fast instead of stalling until the server launch timeout.
|
||||
batch_input_ids = [
|
||||
[1, 2, 3, 4, 5],
|
||||
[10, 11, 12, 13, 14],
|
||||
[20, 21, 22, 23, 24],
|
||||
[30, 31, 32, 33, 34],
|
||||
]
|
||||
response = requests.post(
|
||||
self.base_url + "/generate",
|
||||
json={
|
||||
"input_ids": batch_input_ids,
|
||||
"sampling_params": {"max_new_tokens": 8, "temperature": 0},
|
||||
},
|
||||
timeout=60,
|
||||
)
|
||||
self.assertEqual(response.status_code, 200, response.text)
|
||||
results = response.json()
|
||||
# Every batched request must get its reply routed back — not hang.
|
||||
self.assertEqual(len(results), len(batch_input_ids))
|
||||
for result in results:
|
||||
self.assertIn("text", result)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user