Skip to content

Commit e0bb33c

Browse files
authored
move accelerator_connector.py to the connectors subfolder (#6033)
* move accelerator connector * rename BackendConnector -> AcceleratorConnector
1 parent e01446c commit e0bb33c

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

pytorch_lightning/accelerators/accelerator_connector.py renamed to pytorch_lightning/trainer/connectors/accelerator_connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
import horovod.torch as hvd
6161

6262

63-
class BackendConnector(object):
63+
class AcceleratorConnector(object):
6464

6565
def __init__(
6666
self,

pytorch_lightning/trainer/deprecated_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14-
from pytorch_lightning.accelerators.accelerator_connector import BackendConnector
14+
from pytorch_lightning.trainer.connectors.accelerator_connector import AcceleratorConnector
1515
from pytorch_lightning.trainer.states import RunningStage
1616
from pytorch_lightning.utilities import DeviceType, DistributedType, rank_zero_warn
1717

@@ -22,7 +22,7 @@ class DeprecatedDistDeviceAttributes:
2222
_device_type: DeviceType
2323
_running_stage: RunningStage
2424
num_gpus: int
25-
accelerator_connector: BackendConnector
25+
accelerator_connector: AcceleratorConnector
2626

2727
@property
2828
def on_cpu(self) -> bool:

pytorch_lightning/trainer/properties.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from torch.optim import Optimizer
2222

2323
from pytorch_lightning.accelerators import Accelerator
24-
from pytorch_lightning.accelerators.accelerator_connector import BackendConnector
24+
from pytorch_lightning.trainer.connectors.accelerator_connector import AcceleratorConnector
2525
from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint, ProgressBarBase
2626
from pytorch_lightning.callbacks.base import Callback
2727
from pytorch_lightning.core.lightning import LightningModule
@@ -51,7 +51,7 @@ class TrainerProperties(ABC):
5151
_state: TrainerState
5252
_weights_save_path: str
5353

54-
accelerator_connector: BackendConnector
54+
accelerator_connector: AcceleratorConnector
5555
callbacks: List[Callback]
5656
checkpoint_connector: CheckpointConnector
5757
limit_val_batches: int

pytorch_lightning/trainer/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from pytorch_lightning import _logger as log
2424
from pytorch_lightning.accelerators import Accelerator
25-
from pytorch_lightning.accelerators.accelerator_connector import BackendConnector
25+
from pytorch_lightning.trainer.connectors.accelerator_connector import AcceleratorConnector
2626
from pytorch_lightning.callbacks import Callback
2727
from pytorch_lightning.core.datamodule import LightningDataModule
2828
from pytorch_lightning.core.lightning import LightningModule
@@ -308,7 +308,7 @@ def __init__(
308308
self.data_connector = DataConnector(self)
309309
self.optimizer_connector = OptimizerConnector(self)
310310

311-
self.accelerator_connector = BackendConnector(
311+
self.accelerator_connector = AcceleratorConnector(
312312
num_processes, tpu_cores, distributed_backend, auto_select_gpus, gpus, num_nodes, sync_batchnorm, benchmark,
313313
replace_sampler_ddp, deterministic, precision, amp_backend, amp_level, plugins
314314
)

0 commit comments

Comments
 (0)