fix linear recomposition

mcr229 · facebook-github-bot · commit be125a453931 · 2024-04-16T13:00:06.000-07:00
Summary: Fixes the torchat ci where we are failing with expand copy.

Reviewed By: digantdesai

Differential Revision: D56204667
diff --git a/backends/xnnpack/passes/convert_to_linear.py b/backends/xnnpack/passes/convert_to_linear.py
@@ -36,6 +36,7 @@ class ConvertToLinearPass(XNNPACKPass):
     targets = [
         exir_ops.edge.aten.mm.default,
         exir_ops.edge.aten.addmm.default,
+        exir_ops.edge.aten.bmm.default,
     ]
 
     @staticmethod

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@ class ConvertToLinearPass(XNNPACKPass):`
`36`	`36`	`targets = [`
`37`	`37`	`exir_ops.edge.aten.mm.default,`
`38`	`38`	`exir_ops.edge.aten.addmm.default,`
	`39`	`+ exir_ops.edge.aten.bmm.default,`
`39`	`40`	`]`
`40`	`41`
`41`	`42`	`@staticmethod`