Update torch to 2.5.1

pytorch · Dec 4, 2024 · 24f24de · 24f24de
1 parent 41cbf47
commit 24f24de
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/experimental/torch_xla2/dev-requirements.txt b/experimental/torch_xla2/dev-requirements.txt
@@ -1,4 +1,4 @@
 -f https://download.pytorch.org/whl/torch
-torch==2.4.0; sys_platform == 'darwin'  # macOS
-torch==2.4.0+cpu; sys_platform != 'darwin' # Non-macOS (CPU-only), like on TPU
+torch==2.5.1; sys_platform == 'darwin'  # macOS
+torch==2.5.1+cpu; sys_platform != 'darwin' # Non-macOS (CPU-only), like on TPU
 ruff~=0.3.5
diff --git a/experimental/torch_xla2/torch_xla2/export.py b/experimental/torch_xla2/torch_xla2/export.py
@@ -169,9 +169,10 @@ def _build_symbolic_constraints(symbol_name, torch_constraint):
       symbol = sympy.Symbol(symbol_name)
       if torch_constraint.lower != 2:
         constraints.append(symbol >= torch_constraint.lower)
-      if not torch_constraint.upper.is_infinite:
+      from sympy.core.singleton import S
+      if not torch_constraint.upper.is_infinite and torch_constraint.upper is not S.IntInfinity:
         constraints.append(symbol <= torch_constraint.upper)
-      
+
       return tuple(sympy.pretty(c, use_unicode=False) for c in constraints)
 
     def _build_symbolic_shape(sym, constraint, free_symbols):