From 9c2dd5775a59fc6a08d8509193294846dcdce73f Mon Sep 17 00:00:00 2001 From: Horace He Date: Fri, 5 Feb 2021 11:23:42 -0800 Subject: [PATCH] Fixed slight bug in FX docs (#51779) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51779 Reviewed By: ngimel Differential Revision: D26279623 Pulled By: Chillee fbshipit-source-id: 0cd2a487ce6b80ce0d3f81e2b2334ade20d816bb --- docs/source/fx.rst | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/fx.rst b/docs/source/fx.rst index cf8890f16bf..c779f30aaa0 100644 --- a/docs/source/fx.rst +++ b/docs/source/fx.rst @@ -269,6 +269,7 @@ on them and append them to the :class:`Graph`. """ graph : fx.Graph = tracer_class().trace(model) new_graph = fx.Graph() + env = {} for node in graph.nodes: if node.op == 'call_function' and node.target in decomposition_rules: # By wrapping the arguments with proxies,