File tree Expand file tree Collapse file tree 5 files changed +1
-5
lines changed
src/lightning/fabric/accelerators Expand file tree Collapse file tree 5 files changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -63,4 +63,4 @@ def name() -> str:
6363
6464 @classmethod
6565 def register_accelerators (cls , accelerator_registry : _AcceleratorRegistry ) -> None :
66- accelerator_registry .register (cls .name (), cls , description = cls .__name__ )
66+ accelerator_registry .register (cls .name (), cls , description = cls .__name__ )
Original file line number Diff line number Diff line change 1717from typing_extensions import override
1818
1919from lightning .fabric .accelerators .accelerator import Accelerator
20- from lightning .fabric .accelerators .registry import _AcceleratorRegistry
2120
2221
2322class CPUAccelerator (Accelerator ):
Original file line number Diff line number Diff line change 1818from typing_extensions import override
1919
2020from lightning .fabric .accelerators .accelerator import Accelerator
21- from lightning .fabric .accelerators .registry import _AcceleratorRegistry
2221from lightning .fabric .utilities .rank_zero import rank_zero_info
2322
2423
Original file line number Diff line number Diff line change 2020from typing_extensions import override
2121
2222from lightning .fabric .accelerators .accelerator import Accelerator
23- from lightning .fabric .accelerators .registry import _AcceleratorRegistry
2423
2524
2625class MPSAccelerator (Accelerator ):
Original file line number Diff line number Diff line change 1919from typing_extensions import override
2020
2121from lightning .fabric .accelerators .accelerator import Accelerator
22- from lightning .fabric .accelerators .registry import _AcceleratorRegistry
2322from lightning .fabric .utilities .device_parser import _check_data_type
2423
2524
You can’t perform that action at this time.
0 commit comments