remove the None branch for get_current_accelerator_device

zxd1997066 · zxd1997066 · commit b3fb633a3643 · 2026-01-13T11:31:27.000+08:00
diff --git a/torchao/utils.py b/torchao/utils.py
@@ -155,8 +155,6 @@ def get_available_devices():
 def get_current_accelerator_device():
     if torch.accelerator.is_available():
         return torch.accelerator.current_accelerator()
-    else:
-        return None
 
 
 def get_compute_capability():
