Skip to content

Commit

Permalink
Allowing FSDP strategy for hpu accelerator with strategy flag set
Browse files Browse the repository at this point in the history
Signed-off-by: jyothi kumar sambolu <jsambolu@habana.ai>
  • Loading branch information
jyothisambolu committed Mar 27, 2024
1 parent f7fe623 commit 995aaaa
Showing 1 changed file with 2 additions and 0 deletions.
Expand Up @@ -460,6 +460,8 @@ def _check_strategy_and_fallback(self) -> None:
if _habana_available_and_importable():
from lightning_habana import HPUAccelerator
if isinstance(self._accelerator_flag, HPUAccelerator):
if strategy_flag:
self._strategy_flag = strategy_flag
return
raise MisconfigurationException(
f"You selected strategy to be `{FSDPStrategy.strategy_name}`, but GPU accelerator is not used."
Expand Down

0 comments on commit 995aaaa

Please sign in to comment.