fix linear recomposition (#3064)

mcr229 · facebook-github-bot · commit 4b6d2c3930e8 · 2024-04-16T16:38:11.000-07:00
Summary: Pull Request resolved: #3064 Fixes the torchat ci where we are failing with expand copy. Reviewed By: digantdesai, mikekgfb, kirklandsign Differential Revision: D56204667 fbshipit-source-id: 1d648460b59785884c33cdd479eb9c4c7d452a2a
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`