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,