fixes fastai#3981 to stop deprecated warning messages, use the curren…

…t torch.backends.mps.is_built attribute
hertera1 · Dec 15, 2023 · 2e4beb1 · 2e4beb1
1 parent 7723c4e
commit 2e4beb1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastai/torch_core.py b/fastai/torch_core.py
@@ -260,7 +260,7 @@ def to_float(b):
 # %% ../nbs/00_torch_core.ipynb 71
 def _has_mps():
     if nested_attr(torch, 'backends.mps.is_available', noop)(): return True
-    return getattr(torch, 'has_mps', False)
+    return nested_attr(torch, 'backends.mps.is_built', False)()
 
 def default_device(use=-1):
     "Return or set default device; `use_cuda`: -1 - CUDA/mps if available; True - error if not available; False - CPU"

diff --git a/nbs/00_torch_core.ipynb b/nbs/00_torch_core.ipynb
@@ -960,7 +960,7 @@
     "#|export\n",
     "def _has_mps():\n",
     "    if nested_attr(torch, 'backends.mps.is_available', noop)(): return True\n",
-    "    return getattr(torch, 'has_mps', False)\n",
+    "    return nested_attr(torch, 'backends.mps.is_built', False)()\n",
     "\n",
     "def default_device(use=-1):\n",
     "    \"Return or set default device; `use_cuda`: -1 - CUDA/mps if available; True - error if not available; False - CPU\"\n",