File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
pytorch/trainer/connectors Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 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
3333from lightning .fabric .utilities .imports import _IS_INTERACTIVE
3434from lightning .pytorch .accelerators import AcceleratorRegistry
3535from 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 :
You can’t perform that action at this time.
0 commit comments