Unverified Commit 5fff4406 authored by Kevin McKay's avatar Kevin McKay Committed by GitHub
Browse files

[Bugfix] Replace BaseException with specific exceptions in FLA utils (#31590)


Signed-off-by: default avatarc0de128 <kevin.mckay@outlook.com>
parent 1f5b7c41
......@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment