File tree Expand file tree Collapse file tree 4 files changed +0
-4
lines changed
src/lightning/pytorch/accelerators Expand file tree Collapse file tree 4 files changed +0
-4
lines changed Original file line number Diff line number Diff line change 1717from lightning_utilities .core .imports import RequirementCache
1818from typing_extensions import override
1919
20- from lightning .fabric .accelerators import _AcceleratorRegistry
2120from lightning .fabric .accelerators .cpu import _parse_cpu_cores
2221from lightning .fabric .utilities .types import _DEVICE
2322from lightning .pytorch .accelerators .accelerator import Accelerator
Original file line number Diff line number Diff line change 2121from typing_extensions import override
2222
2323import lightning .pytorch as pl
24- from lightning .fabric .accelerators import _AcceleratorRegistry
2524from lightning .fabric .accelerators .cuda import _check_cuda_matmul_precision , _clear_cuda_memory , num_cuda_devices
2625from lightning .fabric .utilities .device_parser import _parse_gpu_ids
2726from lightning .fabric .utilities .types import _DEVICE
Original file line number Diff line number Diff line change 1616import torch
1717from typing_extensions import override
1818
19- from lightning .fabric .accelerators import _AcceleratorRegistry
2019from lightning .fabric .accelerators .mps import MPSAccelerator as _MPSAccelerator
2120from lightning .fabric .utilities .device_parser import _parse_gpu_ids
2221from lightning .fabric .utilities .types import _DEVICE
Original file line number Diff line number Diff line change 1515
1616from typing_extensions import override
1717
18- from lightning .fabric .accelerators import _AcceleratorRegistry
1918from lightning .fabric .accelerators .xla import XLAAccelerator as FabricXLAAccelerator
2019from lightning .fabric .utilities .types import _DEVICE
2120from lightning .pytorch .accelerators .accelerator import Accelerator
You can’t perform that action at this time.
0 commit comments