Skip to content

Commit 456ed6a

Browse files
authored
Disable ReplaceConvWithChannelLastConvPass.
Differential Revision: D75560675 Pull Request resolved: #11211
1 parent ac866ce commit 456ed6a

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)