Merge pull request #3994 from jkyamog/fix_for_mps_deprecated_warning

fixes #3981 to stop deprecated warning messages
fastai · Jan 7, 2024 · 43dbef3 · 43dbef3
2 parents 7723c4e + 2e4beb1
commit 43dbef3
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",