Skip to content

Commit 85d6e55

Browse files
cccclaifacebook-github-bot
authored andcommitted
Patch the _is_conv_node function
Summary: X-link: pytorch/pytorch#153749 torch.ops.aten.conv2d.padding is also conv2d node Reviewed By: andrewor14, jerryzh168 Differential Revision: D74898941
1 parent 5549da8 commit 85d6e55

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchao/quantization/pt2e/utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -625,8 +625,11 @@ def _is_conv_node(n: Node):
625625
"""
626626
return n.op == "call_function" and n.target in [
627627
torch.ops.aten.conv1d.default,
628+
torch.ops.aten.conv1d.padding,
628629
torch.ops.aten.conv2d.default,
630+
torch.ops.aten.conv2d.padding,
629631
torch.ops.aten.conv3d.default,
632+
torch.ops.aten.conv3d.padding,
630633
]
631634

632635

0 commit comments

Comments
 (0)