We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 14ce8f9 commit 1d0766aCopy full SHA for 1d0766a
torchao/utils.py
@@ -153,10 +153,8 @@ def get_available_devices():
153
154
155
def get_current_accelerator_device():
156
- if torch.accelerator.is_available():
157
- return torch.accelerator.current_accelerator()
158
- else:
159
- return None
+ assert torch.accelerator.is_available()
+ return torch.accelerator.current_accelerator()
160
161
162
def get_compute_capability():
0 commit comments