nit

BoyuanFeng · BoyuanFeng · commit 7586fd60b79c · 2025-06-02T20:01:00.000-07:00
diff --git a/torch/_inductor/scheduler.py b/torch/_inductor/scheduler.py
@@ -4254,11 +4254,8 @@ def filter_symbols(
         res: OrderedSet[sympy.Symbol] = OrderedSet()
         for s in candidate_symbols:
             symplified_s = V.graph.sizevars.simplify(s)
-            # skip when s is simplified to an Integer
-            if isinstance(symplified_s, sympy.Symbol):
-                res.add(symplified_s)
-            else:
-                assert isinstance(symplified_s, sympy.Integer)
+            # use free_symbols only when s is simplified to an Integer or expr
+            res.update(symplified_s.free_symbols)
 
         return res
 
