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 59b6935 commit 2c62b3fCopy full SHA for 2c62b3f
src/lightning/fabric/utilities/device_parser.py
@@ -183,6 +183,7 @@ def _get_all_available_gpus(
183
xpu_gpus = []
184
if _lightning_xpu_available():
185
from lightning_xpu.fabric import _get_all_visible_xpu_devices
186
+
187
xpu_gpus += _get_all_visible_xpu_devices() if include_xpu else []
188
return cuda_gpus + mps_gpus + xpu_gpus
189
src/lightning/fabric/utilities/imports.py
@@ -17,7 +17,7 @@
17
import platform
18
import sys
19
20
-from lightning_utilities.core.imports import compare_version, RequirementCache
+from lightning_utilities.core.imports import RequirementCache, compare_version
21
22
_IS_WINDOWS = platform.system() == "Windows"
23
0 commit comments