dynamo: Don't crash with internal error if getattr on a tensor fails (pytorch#144817)

This prevents crashes when getattr is called on a tensor for something
which doesn't exist.

Pull Request resolved: pytorch#144817
Approved by: https://github.com/williamwen42, https://github.com/jansel

Author: c00w

test/dynamo/test_compile.py

@@ -175,6 +175,17 @@ def fn(x):
 
         x = torch.randn(10, 10)
         with self.assertRaises(NameError):
+        fn(x)
+
+    def test_compilation_tensor_invalid_method(self):
+        @torch.compile(backend="eager")
+        def fn(x):
+            y = torch.tensor(x)
+            return y.doesnotexist()
+
+        x = torch.randn(10, 10)
+
+        with self.assertRaises(AttributeError):
             fn(x)
 
 

torch/_dynamo/utils.py

@@ -3068,10 +3068,16 @@ def run_node(tracer, node, args, kwargs, nnmodule):
         def make_error_message(e):
             return f"Failed running {op} {node.target}(*{args}, **{kwargs}):\n" + str(e)
 
+        from .exc import Unsupported
+
         try:
             if op == "call_function":
                 return node.target(*args, **kwargs)
             elif op == "call_method":
+                if not hasattr(args[0], node.target):
+                    from .exc import unimplemented
+
+                    unimplemented(make_error_message("attribute not defined"))
                 return getattr(args[0], node.target)(*args[1:], **kwargs)
             elif op == "call_module":
                 assert nnmodule is not None
@@ -3087,6 +3093,8 @@ def make_error_message(e):
             from .exc import unimplemented
 
             unimplemented(make_error_message(e), from_exc=e)
+        except Unsupported:
+            raise
         except Exception as e:
             raise RuntimeError(make_error_message(e)).with_traceback(
                 e.__traceback__