Skip to content

Commit bbae4b1

Browse files
hsharma35facebook-github-bot
authored andcommitted
Disable ReplaceConvWithChannelLastConvPass.
Summary: Does not work in pt2. Differential Revision: D75560675
1 parent e78a944 commit bbae4b1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/cadence/aot/replace_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -939,8 +939,8 @@ def replace_conv_with_nhwc_conv(self, graph_module: torch.fx.GraphModule):
939939

940940
# This pass needs to be reworked to be compatible with PT2. It is an optimization
941941
# pass anyway, so move it to opt level 2.
942-
# TODO(matthiascremon): update and improve this pass.
943-
@register_cadence_pass(CadencePassAttribute(opt_level=2))
942+
# TODO: T213724613 update and improve this pass.
943+
# @register_cadence_pass(CadencePassAttribute(opt_level=2))
944944
class ReplaceConvWithChannelLastConvPass(ExportPass):
945945
"""
946946
Replace the ATen convolution op with custom conv op with NCHW or NHWC layout

0 commit comments

Comments
 (0)