Update fabric imports for HPU strategies

CHANGELOG.md

@@ -5,6 +5,29 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [Unreleased version] - YYYY-MM-DD
+
+### Added
+
+-
+
+### Changed
+
+
+
+### Fixed
+
+- Fixed fabric imports for HPU strategies ([#126](https://github.com/Lightning-AI/lightning-Habana/pull/126))
+
+### Removed
+
+-
+
+### Deprecated
+
+-
+
+
 ## [1.3.0] - 2023-12-06
 
 ### Added
@@ -20,7 +43,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 
--
+- Fixed fabric imports for HPU strategies ([#126](https://github.com/Lightning-AI/lightning-Habana/pull/126))
 
 ### Removed
 

src/lightning_habana/__about__.py

@@ -1,4 +1,4 @@
-__version__ = "1.3.0"
+__version__ = "1.4.0-dev0"
 __author__ = "Lightning-AI et al."
 __author_email__ = "name@lightning.ai"
 __license__ = "Apache-2.0"

src/lightning_habana/fabric/accelerator.py

@@ -18,9 +18,9 @@
 from lightning_utilities import module_available
 
 if module_available("lightning"):
-    from lightning.pytorch.accelerators.accelerator import Accelerator
+    from lightning.fabric.accelerators import Accelerator
 elif module_available("pytorch_lightning"):
-    from pytorch_lightning.accelerators.accelerator import Accelerator
+    from lightning_fabric.accelerators import Accelerator
 else:
     raise ModuleNotFoundError("You are missing `lightning` or `pytorch-lightning` package, please install it.")
 

src/lightning_habana/fabric/strategies/parallel.py

@@ -115,7 +115,7 @@ def determine_ddp_device_ids(self) -> None:
     #     return obj[0]
 
     def backward(self, tensor: Tensor, module: Optional[Module], *args: Any, **kwargs: Any) -> None:
-        super().backward(tensor=Tensor, module=module, args=args, kwargs=kwargs)
+        super().backward(tensor, module=module, args=args, kwargs=kwargs)
         if _TORCH_LESSER_EQUAL_1_13_1:
             # Break lazy accumulation of graph after fwd+bwd
             htcore.mark_step()

src/lightning_habana/fabric/strategies/single.py

@@ -78,7 +78,7 @@ def checkpoint_io(self, io: Optional[CheckpointIO]) -> None:
         self._checkpoint_io = io
 
     def backward(self, tensor: Tensor, module: Optional[Module], *args: Any, **kwargs: Any) -> None:
-        super().backward(tensor=Tensor, module=module, *args, **kwargs)
+        super().backward(tensor, module=module, *args, **kwargs)
         if _TORCH_LESSER_EQUAL_1_13_1:
             # Break lazy accumulation of graph after fwd+bwd
             htcore.mark_step()