diff --git a/src/pytorch_lightning/strategies/hpu_parallel.py b/src/pytorch_lightning/strategies/hpu_parallel.py index 9e0dc84402cca..35f40b9cef4a4 100644 --- a/src/pytorch_lightning/strategies/hpu_parallel.py +++ b/src/pytorch_lightning/strategies/hpu_parallel.py @@ -32,7 +32,7 @@ if _HPU_AVAILABLE: import habana_frameworks.torch.core as htcore - import habana_frameworks.torch.core.hccl # noqa: F401 + import habana_frameworks.torch.distributed.hccl # noqa: F401 log = logging.getLogger(__name__) diff --git a/src/pytorch_lightning/strategies/single_hpu.py b/src/pytorch_lightning/strategies/single_hpu.py index 7190db0061f0a..cbc9dee12640f 100644 --- a/src/pytorch_lightning/strategies/single_hpu.py +++ b/src/pytorch_lightning/strategies/single_hpu.py @@ -24,7 +24,6 @@ if _HPU_AVAILABLE: import habana_frameworks.torch.core as htcore - import habana_frameworks.torch.core.hccl # noqa: F401 class SingleHPUStrategy(SingleDeviceStrategy):