Runtime check CUDA driver version to avoid unresolved green context symbols (#9021)

2025-08-12 09:26:10 -07:00
parent 3a9afe2a42
commit 445f9dca6e
3 changed files with 28 additions and 36 deletions
--- a/sgl-kernel/python/sgl_kernel/utils.py
+++ b/sgl-kernel/python/sgl_kernel/utils.py
@@ -14,6 +14,7 @@
 # ==============================================================================

 import functools
+import subprocess
 from typing import Dict, Tuple

 import torch