Skip to content

Commit 7d1d814

Browse files
committed
fix cpu tests on macos
1 parent cfff0a7 commit 7d1d814

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/lightning/fabric/cli.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -203,14 +203,10 @@ def _get_num_processes(accelerator: str, devices: str) -> int:
203203
elif torch.backends.mps.is_available() and torch.backends.mps.is_built():
204204
parsed_devices = MPSAccelerator.parse_devices(devices)
205205
else:
206-
parsed_devices = CPUAccelerator.parse_devices(devices)
206+
return CPUAccelerator.parse_devices(devices)
207207
else:
208-
parsed_devices = CPUAccelerator.parse_devices(devices)
209-
return (
210-
len(parsed_devices)
211-
if isinstance(parsed_devices, list)
212-
else (parsed_devices if isinstance(parsed_devices, int) else 0)
213-
)
208+
return CPUAccelerator.parse_devices(devices)
209+
return len(parsed_devices) if parsed_devices is not None else 0
214210

215211

216212
def _torchrun_launch(args: Namespace, script_args: list[str]) -> None:

0 commit comments

Comments
 (0)