Update base for Update on "[wip] Improve exception support"

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx chenyang78 kadeng chauhang

[ghstack-poisoned]

torch/_dynamo/variables/tensor.py

@@ -363,7 +363,7 @@ def var_getattr(self, tx, name):
         # For attributes (not methods) that were not caught in the special handling above,
         # (e.g. tensor.real), we handle these generically, assuming that the output type is
         # a tensor.
-        if result is None and name != "grad":
+        if result is None and name not in ("grad", "grad_fn"):
 
             def try_generic_attr_handling():
                 from .builder import wrap_fx_proxy

torch/_dynamo/variables/user_defined.py

@@ -834,6 +834,10 @@ def var_getattr(self, tx, name):
         if tx.output.side_effects.has_pending_mutation_of_attr(self, name):
             return tx.output.side_effects.load_attr(self, name)
 
+        if name == "__dict__":
+            options = {"source": source}
+            return variables.GetAttrVariable(self, name, **options)
+
         try:
             subobj = self._getattr_static(name)
         except AttributeError: