Method replace_by_example

torch/_inductor/pattern_matcher.py:124–136

(self, replacement_fn, args, trace_fn=None)

        )

    def replace_by_example(self, replacement_fn, args, trace_fn=None):
        assert self.ctx
        if trace_fn is None:
            trace_fn = fwd_only
        replacement = trace_fn(
            replacement_fn, torch.fx.map_arg(args, lambda arg: arg.meta["val"])
        )
        ReplacementPatternEntry.replace_with_graph(
            self,
            self.ctx.graph,
            replacement,
            args,
        )


class FailedMatch(RuntimeError):