Skip to content

Commit 1519c66

Browse files
committed
change function name
1 parent 019b3b8 commit 1519c66

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/lightning/fabric/utilities/device_parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def _check_data_type(device_ids: object) -> None:
206206
raise TypeError(f"{msg} {device_ids!r}.")
207207

208208

209-
def _select_auto_accelerator_fabric() -> str:
209+
def _select_auto_accelerator() -> str:
210210
"""Choose the accelerator type (str) based on availability."""
211211
from lightning.fabric.accelerators.cuda import CUDAAccelerator
212212
from lightning.fabric.accelerators.mps import MPSAccelerator

src/lightning/pytorch/trainer/connectors/accelerator_connector.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
SLURMEnvironment,
3030
TorchElasticEnvironment,
3131
)
32-
from lightning.fabric.utilities.device_parser import _determine_root_gpu_device, _select_auto_accelerator_fabric
32+
from lightning.fabric.utilities.device_parser import _determine_root_gpu_device, _select_auto_accelerator
3333
from lightning.fabric.utilities.imports import _IS_INTERACTIVE
3434
from lightning.pytorch.accelerators import AcceleratorRegistry
3535
from lightning.pytorch.accelerators.accelerator import Accelerator
@@ -337,7 +337,7 @@ def _choose_auto_accelerator() -> str:
337337

338338
if HPUAccelerator.is_available():
339339
return "hpu"
340-
return _select_auto_accelerator_fabric()
340+
return _select_auto_accelerator()
341341

342342
@staticmethod
343343
def _choose_gpu_accelerator_backend() -> str:

0 commit comments

Comments
 (0)