File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 2525from lightning .fabric .plugins .precision .precision import _PRECISION_INPUT_STR , _PRECISION_INPUT_STR_ALIAS
2626from lightning .fabric .strategies import STRATEGY_REGISTRY
2727from lightning .fabric .utilities .consolidate_checkpoint import _process_cli_args
28- from lightning .fabric .utilities .device_parser import _parse_gpu_ids , _select_auto_accelerator_fabric
28+ from lightning .fabric .utilities .device_parser import _parse_gpu_ids , _select_auto_accelerator
2929from lightning .fabric .utilities .distributed import _suggested_max_num_threads
3030from lightning .fabric .utilities .load import _load_distributed_checkpoint
3131
@@ -189,7 +189,7 @@ def _get_num_processes(accelerator: str, devices: str) -> int:
189189 """Parse the `devices` argument to determine how many processes need to be launched on the current machine."""
190190
191191 if accelerator == "auto" or accelerator is None :
192- accelerator = _select_auto_accelerator_fabric ()
192+ accelerator = _select_auto_accelerator ()
193193 if devices == "auto" :
194194 if accelerator == "cuda" or accelerator == "mps" or accelerator == "cpu" :
195195 devices = "1"
You can’t perform that action at this time.
0 commit comments