Skip to content

Commit 8787429

Browse files
ethansfngfacebook-github-bot
authored andcommitted
Update ReplaceConvolutionOptionalArgsWithConcreteArgsPass to work with cadence.convolution
Summary: ReplaceConvolutionOptionalArgsWithConcreteArgsPass runs after ReplaceAtenConvolutionWithCadenceConvolutionPass so ReplaceConvolutionOptionalArgsWithConcreteArgsPass should be configured to run on a cadence.convolution not aten.convolution Differential Revision: D82842567
1 parent 641e737 commit 8787429

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
@@ -438,11 +438,11 @@ class ReplaceConvolutionOptionalArgsWithConcreteArgsPass(ExportPass):
438438
"""
439439

440440
def call_operator(self, op, args, kwargs, meta):
441-
if get_edge_overload_packet(op) != exir_ops.edge.aten.convolution:
441+
if get_edge_overload_packet(op) != exir_ops.edge.cadence.convolution:
442442
return super().call_operator(op, args, kwargs, meta)
443443

444444
# Check if the bias is already concrete
445-
assert len(args) == 9
445+
assert len(args) == 8
446446
if args[2] is not None:
447447
return super().call_operator(op, args, kwargs, meta)
448448

0 commit comments

Comments
 (0)