[misc] remove is_cuda_available (#5319)

2025-04-21 09:16:51 +08:00
parent 1195182040
commit 97cb762bb6
14 changed files with 42 additions and 47 deletions
--- a/python/sglang/srt/layers/layernorm.py
+++ b/python/sglang/srt/layers/layernorm.py
@@ -20,9 +20,9 @@ import torch
 import torch.nn as nn

 from sglang.srt.custom_op import CustomOp
-from sglang.srt.utils import is_cuda_available
+from sglang.srt.utils import is_cuda

-_is_cuda = is_cuda_available()
+_is_cuda = is_cuda()

 if _is_cuda:
    from sgl_kernel import (