Update

[ghstack-poisoned]

torch/_dynamo/variables/dicts.py

@@ -1371,3 +1371,13 @@ class DictItemsVariable(DictViewVariable):
 
     def python_type(self):
         return dict_items
+
+    def call_method(self, tx, name, args, kwargs):
+        # TODO(guilhermeleobas): This should actually check if args[0]
+        # implements the mapping protocol.
+        if name == "__eq__":
+            assert len(args) == 1
+            if isinstance(args[0], DictItemsVariable):
+                return self.dv_dict.call_method(tx, "__eq__", [args[0].dv_dict], {})
+            return ConstantVariable.create(False)
+        return super().call_method(tx, name, args, kwargs)

torch/_dynamo/variables/user_defined.py

@@ -1944,8 +1944,13 @@ class UserDefinedDictVariable(UserDefinedObjectVariable):
             assert self.source is None, (
                 "dict_vt must be constructed by builder.py when source is present"
             )
+            user_cls = (
+                collections.OrderedDict
+                if isinstance(value, collections.OrderedDict)
+                else dict
+            )
             self._dict_vt = variables.ConstDictVariable(
-                {}, type(value), mutation_type=ValueMutationNew()
+                {}, user_cls, mutation_type=ValueMutationNew()
             )
         self._dict_methods = dict_methods