[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/quantization/modelopt_quant.py
+++ b/python/sglang/srt/layers/quantization/modelopt_quant.py
@@ -22,9 +22,9 @@ from sglang.srt.layers.quantization.utils import (
    requantize_with_max_scale,
 )
 from sglang.srt.layers.radix_attention import RadixAttention
-from sglang.srt.utils import is_cuda_available
+from sglang.srt.utils import is_cuda

-if is_cuda_available():
+if is_cuda():
    from sgl_kernel import cutlass_scaled_fp4_mm, scaled_fp4_quant

 # Initialize logger for the module