diff --git a/src/lightning/fabric/accelerators/xla.py b/src/lightning/fabric/accelerators/xla.py index d438197329939..07c51ceb38286 100644 --- a/src/lightning/fabric/accelerators/xla.py +++ b/src/lightning/fabric/accelerators/xla.py @@ -107,7 +107,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No def _using_pjrt() -> bool: # `using_pjrt` is removed in torch_xla 2.5 - if _XLA_GREATER_EQUAL_2_5: + if True: from torch_xla import runtime as xr return xr.device_type() is not None