Cache the result of is_blackwell platform check (#10498)

2025-09-15 22:30:28 -07:00
parent 5fe39e85a2
commit b2435be682
2 changed files with 3 additions and 7 deletions
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -167,6 +167,7 @@ is_ampere_with_cuda_12_3 = lambda: _check(8)
 is_hopper_with_cuda_12_3 = lambda: _check(9)


+@lru_cache(maxsize=1)
 def is_blackwell():
    if not is_cuda():
        return False