replace reshape with convert from real tensor

blzheng · blzheng · commit a6ac94da4fda · 2023-01-15T18:39:43.000-08:00
diff --git a/torch/_dynamo/variables/builder.py b/torch/_dynamo/variables/builder.py
@@ -145,7 +145,10 @@ def get_fake_examples(self):
                 not config.dynamic_shapes
                 and self.fake_tensor.shape != self.example.shape
             ):
-                self.fake_tensor = self.fake_tensor.reshape(self.example.shape)
+                converter = torch._subclasses.fake_tensor.FakeTensorConverter()
+                self.fake_tensor = converter.from_real_tensor(
+                    self.fake_tensor.fake_mode, self.example
+                )
             return [self.fake_tensor]
 
     def __len__(self):
