Skip to content

Commit 9908265

Browse files
generatedunixname499836121facebook-github-bot
generatedunixname499836121
authored andcommitted
Revert "Track node mutations (#152389)"
Summary: This reverts commit c2936ebfd58be7a6519f51d165dfac8407020140. Reverted pytorch/pytorch#152389 on behalf of https://github.com/jeanschmidt due to Humm, interesting, there seems to be a bug in stack PRs, as it should be part of the stack and be reverted with the other ones ([comment](pytorch/pytorch#152389 (comment))) Reviewed By: jeanschmidt Differential Revision: D74598324 fbshipit-source-id: 83adbb89c64e0067449a863e1e5beef0a45aeefe
1 parent 595677c commit 9908265

File tree

1 file changed

+0
-15
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+0
-15
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -3150,10 +3150,6 @@ def get_fake_value(node, tx, allow_non_graph_fake=False):
31503150
args, kwargs = get_fake_values_from_nodes(
31513151
tx, (node.args, node.kwargs), allow_non_graph_fake
31523152
)
3153-
flat_args_kwargs, _ = pytree.tree_flatten((args, kwargs))
3154-
id_to_initial_version = {
3155-
id(arg): arg._version for arg in flat_args_kwargs if is_fake(arg)
3156-
}
31573153

31583154
nnmodule = None
31593155
if op == "call_method" and len(args) > 0 and isinstance(args[0], torch.nn.Module):
@@ -3294,17 +3290,6 @@ def get_fake_value(node, tx, allow_non_graph_fake=False):
32943290
_ = pytree.tree_map_only(
32953291
torch.Tensor, functools.partial(ensure_graph_fake, tx=tx), ret_val
32963292
)
3297-
3298-
if (
3299-
torch._dynamo.config.use_graph_deduplication
3300-
or torch._dynamo.config.track_nodes_for_deduplication
3301-
):
3302-
tx.output.region_tracker.track_node_mutations(
3303-
node,
3304-
flat_args_kwargs,
3305-
id_to_initial_version,
3306-
)
3307-
33083293
return ret_val
33093294

33103295

0 commit comments

Comments
 (0)