Skip to content

Commit 4cd25b2

Browse files
committed
combine ipu import
1 parent 8485765 commit 4cd25b2

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

pytorch_lightning/strategies/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from pytorch_lightning.strategies.dp import DataParallelStrategy # noqa: F401
66
from pytorch_lightning.strategies.fully_sharded import DDPFullyShardedStrategy # noqa: F401
77
from pytorch_lightning.strategies.horovod import HorovodStrategy # noqa: F401
8+
from pytorch_lightning.strategies.ipu import IPUStrategy # noqa: F401
89
from pytorch_lightning.strategies.parallel import ParallelStrategy # noqa: F401
910
from pytorch_lightning.strategies.sharded import DDPShardedStrategy # noqa: F401
1011
from pytorch_lightning.strategies.sharded_spawn import DDPSpawnShardedStrategy # noqa: F401

pytorch_lightning/trainer/connectors/accelerator_connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,11 @@
5757
DDPStrategy,
5858
DeepSpeedStrategy,
5959
HorovodStrategy,
60+
IPUStrategy,
6061
SingleDeviceStrategy,
6162
SingleTPUStrategy,
6263
TPUSpawnStrategy,
6364
)
64-
from pytorch_lightning.strategies.ipu import IPUStrategy
6565
from pytorch_lightning.utilities import (
6666
_AcceleratorType,
6767
_StrategyType,

0 commit comments

Comments
 (0)