Update

isuruf · isuruf · commit d71d51a95a2f · 2024-09-11T04:28:23.000Z
[ghstack-poisoned]
diff --git a/torch/fx/experimental/symbolic_shapes.py b/torch/fx/experimental/symbolic_shapes.py
@@ -437,6 +437,8 @@ def integer_coefficient(x):
         if x.is_Integer:
             return abs(int(x))
         elif x.is_Mul:
+            # If one of the args of a Mul is an Integer, it is the
+            # first arg. eg: args(2*x*3*y) == (6, x, y)
             return abs(int(x.args[0])) if x.args[0].is_Integer else 1
         else:
             return 1
