Check if the current compiled version of pytorch supports MPS (#30664)

09edd77f · jiaqianjing · GitHub · 307f632b · 09edd77f
Unverified Commit 09edd77f authored May 06, 2024 by jiaqianjing Committed by GitHub May 06, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/utils/import_utils.py src/transformers/utils/import_utils.py +1 -1

No files found.
--- a/src/transformers/utils/import_utils.py
+++ b/src/transformers/utils/import_utils.py
@@ -373,7 +373,7 @@ def is_torch_mps_available():
        import torch

        if hasattr(torch.backends, "mps"):
-            return torch.backends.mps.is_available()
+            return torch.backends.mps.is_available() and torch.backends.mps.is_built()
    return False