-
Notifications
You must be signed in to change notification settings - Fork 2.1k
关于实时模式的vllm解码并发问题 #3732
Copy link
Copy link
Open
Labels
benchmarkPerformance, accuracy, migration, or benchmark reportPerformance, accuracy, migration, or benchmark reportneeds feedbackWaiting for reporter feedback or retest resultsWaiting for reporter feedback or retest resultsquestionFurther information is requestedFurther information is requested
Description
Activity
Metadata
Metadata
Assignees
Labels
benchmarkPerformance, accuracy, migration, or benchmark reportPerformance, accuracy, migration, or benchmark reportneeds feedbackWaiting for reporter feedback or retest resultsWaiting for reporter feedback or retest resultsquestionFurther information is requestedFurther information is requested
【realtime_ws】并发 streaming session 的vllm解码
测试中并发 streaming session 增多时,首条结果延迟明显上升,超过一定并发后结果趋于停滞。
在约 15+ 个并发 session 下,首字延迟 p50 从 10 个 session 时的 ~1.5s 涨到 ~2.2s(p95
到 ~66s),并随并发继续恶化。以上数值仅供参考,不代表精确基准。
似乎RealtimeBatchingEngine把并发 session 的请求统一汇到单个 worker 线程,底层调用同步 `LLM.generate。实时流式里每个 session 的 handle_client循环要等自己的 partial 返回之后才发下一帧(run_session_work → asyncio.to_thread →阻塞 generate),语音请求在vad处理后请求天然稀疏,batch_wait_ms=10ms的攒批窗口几乎凑不齐。于是请求稀疏 → 批凑不满 → 单条串行 → 客户端等更久 → 更稀疏,形成自锁,吞吐受限。这是我的推测解释,未必准确,供参考。
目前benchmark 文档只到 8 clients,若未来有意支持更高并发,是否可以考虑把解码器切到 AsyncLLMEngine(vLLM v1 的vllm.v1.engine.async_llm.AsyncLLM,同样 enable_prompt_embeds=True),每个 session 的partial / 锁句作独立 request_id提交:引擎按 step 在卡上合批、同时把每条结果返回给各自调用方,天然保住每个 session 的顺序与 partial 中间态,吞吐也随之恢复。是否值得,请各位大佬按定位评估;我这边物理条件有限,测试结果未必准确,也很难支持更高并发下的吞吐测试,若感兴趣可在高并发下尝试验证。