diff --git a/optimum/fx/parallelization/decomp.py b/optimum/fx/parallelization/decomp.py index 312249253e7..7ba18f43438 100644 --- a/optimum/fx/parallelization/decomp.py +++ b/optimum/fx/parallelization/decomp.py @@ -63,7 +63,7 @@ def create_node_(*args, **kwargs): class DecompTracer(GraphAppendingTracer): """ DecompTracer is a tracer class which works together with `DecompositionInterpreter`, it keeps track of tensors and their - corresponding proxy objects during execution process. When invoked with `create_proxy`, it will creates a node in the containing + corresponding proxy objects during execution process. When invoked with `create_proxy`, it creates a node in the containing graph and associate the output tensor of the node with the created proxy. See https://github.com/pytorch/pytorch/blob/main/torch/fx/experimental/proxy_tensor.py for more details.