[misc] remove is_cuda_available (#5319)
This commit is contained in:
@@ -28,9 +28,9 @@ from sglang.srt.distributed import (
|
||||
get_tensor_model_parallel_world_size,
|
||||
)
|
||||
from sglang.srt.layers.quantization.base_config import QuantizationConfig
|
||||
from sglang.srt.utils import is_cuda_available, set_weight_attrs
|
||||
from sglang.srt.utils import is_cuda, set_weight_attrs
|
||||
|
||||
_is_cuda = is_cuda_available()
|
||||
_is_cuda = is_cuda()
|
||||
|
||||
if _is_cuda:
|
||||
from sgl_kernel import gelu_and_mul, gelu_tanh_and_mul, silu_and_mul
|
||||
|
||||
Reference in New Issue
Block a user