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 af34157 commit 39b7d83Copy full SHA for 39b7d83
src/lightning/fabric/utilities/device_parser.py
@@ -182,6 +182,7 @@ def _get_all_available_gpus(
182
xpu_gpus = []
183
if _lightning_xpu_available():
184
from lightning_xpu.fabric import _get_all_visible_xpu_devices
185
+
186
xpu_gpus += _get_all_visible_xpu_devices() if include_xpu else []
187
return cuda_gpus + mps_gpus + xpu_gpus
188
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