From 5fff44064b38012d5675d54620dc3dc3c28eb522 Mon Sep 17 00:00:00 2001 From: Kevin McKay Date: Thu, 1 Jan 2026 21:27:54 -0600 Subject: [PATCH] [Bugfix] Replace BaseException with specific exceptions in FLA utils (#31590) Signed-off-by: c0de128 --- vllm/model_executor/layers/fla/ops/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/fla/ops/utils.py b/vllm/model_executor/layers/fla/ops/utils.py index 5a48e56a5..18e17a511 100644 --- a/vllm/model_executor/layers/fla/ops/utils.py +++ b/vllm/model_executor/layers/fla/ops/utils.py @@ -119,7 +119,7 @@ def input_guard(fn: Callable[..., torch.Tensor]) -> Callable[..., torch.Tensor]: def get_available_device() -> str: try: return triton.runtime.driver.active.get_current_target().backend - except BaseException: + except (RuntimeError, AttributeError): return "cpu"