Optimize dedupe region ancestor tracking (#152589)

Summary:
X-link: pytorch/pytorch#152589
Approved by: https://github.com/anijain2305
ghstack dependencies: #152389, #152505, #152410, #152506, #152570, #152572

Reviewed By: jeanschmidt

Differential Revision: D74668984

fbshipit-source-id: cdb0ee61902bcde560b369fc065ab5b4d76e6157

Author: mlazos

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -3152,12 +3152,20 @@ def get_fake_value(node, tx, allow_non_graph_fake=False):
     args, kwargs = get_fake_values_from_nodes(
         tx, (node.args, node.kwargs), allow_non_graph_fake
     )
-    flat_args_kwargs = get_fake_values_from_nodes(
-        tx, _get_flat_args(node, {}), allow_non_graph_fake
-    )
-    id_to_initial_version = {
-        id(arg): arg._version for arg in flat_args_kwargs if is_fake(arg)
-    }
+
+    if (
+        torch._dynamo.config.use_graph_deduplication
+        or torch._dynamo.config.track_nodes_for_deduplication
+    ):
+        flat_args_kwargs = get_fake_values_from_nodes(
+            tx, _get_flat_args(node, {}), allow_non_graph_fake
+        )
+        id_to_initial_version = {
+            id(arg): arg._version for arg in flat_args_kwargs if is_fake(arg)
+        }
+    else:
+        flat_args_kwargs = []
+        id_to_initial_version = {}
 
     nnmodule = None
     if op == "call_method" and len(args) > 0 and isinstance(args[0], torch.nn.Module):