I am trying to write custom subgraph rewrite using torch fx subgraph_rewriter. I see that after the replacement the added nodes do not contain “meta” information. Is there any way to capture these during the pattern matching and replace?
you mean “meta” info by modules’s attributes? Like Conv2d’s kernel size, strides, padding etc.