Skip to content

Commit

Permalink
also for Fabric
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Jun 5, 2024
1 parent 97c83e3 commit f90a79a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/lightning/fabric/connector.py
Original file line number Diff line number Diff line change
Expand Up @@ -312,7 +312,8 @@ def _check_device_config_and_set_final_flags(self, devices: Union[List[int], str
f" using {accelerator_name} accelerator."
)

def _choose_auto_accelerator(self) -> str:
@staticmethod
def _choose_auto_accelerator() -> str:
"""Choose the accelerator type (str) based on availability when ``accelerator='auto'``."""
if XLAAccelerator.is_available():
return "tpu"
Expand Down

0 comments on commit f90a79a

Please sign in to comment.