iter.acc

shink · shink · commit 32f625c91f8b · 2025-04-28T07:56:28.000Z
diff --git a/torch/_dynamo/variables/iter.py b/torch/_dynamo/variables/iter.py
@@ -107,8 +107,14 @@ def call_function(
                     gb_type="Unsupported arguments for itertools.accumulate",
                     context=f"call_function {self} {args} {kwargs}",
                     explanation="Dynamo does not know how to trace "
-                    f"itertools.accumulate with args: {args}",
-                    hints=[*graph_break_hints.USER_ERROR],
+                    f"itertools.accumulate with args: {args} and kwargs: {kwargs}. "
+                    "itertools.accumulate expects an iterable, an optional "
+                    "binary function for accumulation, and an optional initial "
+                    "value to set the starting state.",
+                    hints=[
+                        "Make sure the arguments to itertools.accumulate are correct.",
+                        *graph_break_hints.SUPPORTABLE,
+                    ],
                 )
 
             items = []
