Convert invalid args into graph breaks (#121784)

Summary: X-link: pytorch/pytorch#121784 Approved by: https://github.com/yanboliang Reviewed By: jeanschmidt Differential Revision: D54970023 Pulled By: jansel fbshipit-source-id: 79c188aeb2a86189b45a6c56de31ac9d9bd6a318
pytorch · Mar 18, 2024 · 5a6d0dc · 5a6d0dc
1 parent a762495
commit 5a6d0dc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -1729,6 +1729,9 @@ def get_fake_value(node, tx, allow_non_graph_fake=False):
             )
         elif isinstance(cause, ValueRangeError):
             raise UserError(UserErrorType.CONSTRAINT_VIOLATION, e.args[0]) from e
+        elif isinstance(cause, TypeError) and "argument" in str(cause):
+            unimplemented(f"TypeError {node.target}: {cause}")
+
         raise TorchRuntimeError(str(e)).with_traceback(e.__traceback__) from None
 
     if not allow_non_graph_fake: