We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 46291a9 commit 8683ca1Copy full SHA for 8683ca1
build/utils.py
@@ -227,8 +227,8 @@ def is_mps_available() -> bool:
227
# out system says mps is available, but it's not on VMs
228
# so let's set up some memry, and see if that work:
229
try:
230
- mps_tensor = torch.zero(1024, dtype=torch.float16, device="mps")
231
- except:
+ mps_tensor = torch.zeros(1024, dtype=torch.float16, device="mps")
+ except RuntimeError:
232
return False
233
234
# MPS, is that you?
0 commit comments