File tree Expand file tree Collapse file tree 1 file changed +15
-5
lines changed
src/lightning/pytorch/accelerators Expand file tree Collapse file tree 1 file changed +15
-5
lines changed Original file line number Diff line number Diff line change 10
10
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11
11
# See the License for the specific language governing permissions and
12
12
# limitations under the License.
13
+
14
+ __all__ = [
15
+ "Accelerator" ,
16
+ "CPUAccelerator" ,
17
+ "CUDAAccelerator" ,
18
+ "MPSAccelerator" ,
19
+ "XLAAccelerator" ,
20
+ "find_usable_cuda_devices" ,
21
+ ]
22
+
13
23
import sys
14
24
15
- from lightning .fabric .accelerators import find_usable_cuda_devices # noqa: F401
25
+ from lightning .fabric .accelerators import find_usable_cuda_devices
16
26
from lightning .fabric .accelerators .registry import _AcceleratorRegistry
17
27
from lightning .fabric .utilities .registry import _register_classes
18
28
from lightning .pytorch .accelerators .accelerator import Accelerator
19
- from lightning .pytorch .accelerators .cpu import CPUAccelerator # noqa: F401
20
- from lightning .pytorch .accelerators .cuda import CUDAAccelerator # noqa: F401
21
- from lightning .pytorch .accelerators .mps import MPSAccelerator # noqa: F401
22
- from lightning .pytorch .accelerators .xla import XLAAccelerator # noqa: F401
29
+ from lightning .pytorch .accelerators .cpu import CPUAccelerator
30
+ from lightning .pytorch .accelerators .cuda import CUDAAccelerator
31
+ from lightning .pytorch .accelerators .mps import MPSAccelerator
32
+ from lightning .pytorch .accelerators .xla import XLAAccelerator
23
33
24
34
AcceleratorRegistry = _AcceleratorRegistry ()
25
35
_register_classes (AcceleratorRegistry , "register_accelerators" , sys .modules [__name__ ], Accelerator )
You can’t perform that action at this time.
0 commit comments