[Bug] fix green context's incompatibility with cuda < 12.4 (#8701)

2025-08-03 06:23:11 +08:00
parent 6d4fd8826e
commit 603f5ce020
2 changed files with 22 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/spatial.py
+++ b/sgl-kernel/python/sgl_kernel/spatial.py
@@ -14,6 +14,11 @@ def create_greenctx_stream_by_value(
    Returns:
        tuple[ExternalStream, ExternalStream]: The two streams.
    """
+    if torch.version.cuda < "12.4":
+        raise RuntimeError(
+            "Green Contexts feature requires CUDA Toolkit 12.4 or newer."
+        )
+
    if device_id is None:
        device_id = torch.cuda.current_device()