update

justusschock · justusschock · commit ba3271297466 · 2025-11-11T11:58:44.000Z
diff --git a/src/lightning/fabric/accelerators/xla.py b/src/lightning/fabric/accelerators/xla.py
@@ -24,6 +24,8 @@
 from lightning.fabric.utilities.imports import _raise_enterprise_not_available
 
 _XLA_AVAILABLE = RequirementCache("torch_xla>=1.13", "torch_xla")
+_XLA_GREATER_EQUAL_2_1 = RequirementCache("torch_xla>=2.1")
+_XLA_GREATER_EQUAL_2_5 = RequirementCache("torch_xla>=2.5")
 
 
 class XLAAccelerator(Accelerator):
