From abf4f1a47a0769a3fb61d021fd59b797319caa2c Mon Sep 17 00:00:00 2001 From: R0CKSTAR Date: Thu, 26 Mar 2026 11:59:51 +0800 Subject: [PATCH] [MPS] Add StreamContext stub (#20782) Signed-off-by: Xiaodong Ye --- python/sglang/_mps_stub.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/sglang/_mps_stub.py b/python/sglang/_mps_stub.py index aaaebe27e..8a090b398 100644 --- a/python/sglang/_mps_stub.py +++ b/python/sglang/_mps_stub.py @@ -69,6 +69,19 @@ class Event: return 0.0 +class StreamContext: + """Minimal stand-in for ``torch.cuda.StreamContext``.""" + + def __init__(self, stream: Any = None) -> None: + pass + + def __enter__(self) -> "StreamContext": + return self + + def __exit__(self, *args: Any) -> None: + pass + + _default_stream = Stream() @@ -235,6 +248,7 @@ def install() -> None: # Only patch attributes that are actually missing for name, obj in [ ("Stream", Stream), + ("StreamContext", StreamContext), ("Event", Event), ("current_stream", current_stream), ("stream", stream),