Patch the _is_conv_node function

cccclai · facebook-github-bot · commit 110aee3aedcb · 2025-05-16T12:00:15.000-07:00
Summary: torch.ops.aten.conv2d.padding is also conv2d node

Differential Revision: D74898941
diff --git a/torch/ao/quantization/pt2e/utils.py b/torch/ao/quantization/pt2e/utils.py
@@ -168,6 +168,7 @@ def _is_conv_node(n: Node):
     return n.op == "call_function" and n.target in [
         torch.ops.aten.conv1d.default,
         torch.ops.aten.conv2d.default,
+        torch.ops.aten.conv2d.padding,
     ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -168,6 +168,7 @@ def _is_conv_node(n: Node):`
`168`	`168`	`return n.op == "call_function" and n.target in [`
`169`	`169`	`torch.ops.aten.conv1d.default,`
`170`	`170`	`torch.ops.aten.conv2d.default,`
	`171`	`+ torch.ops.aten.conv2d.padding,`
`171`	`172`	`]`
`172`	`173`
`173`	`174`