From be125a45393177bc2de27c515c6f1e38d518f2a9 Mon Sep 17 00:00:00 2001 From: Max Ren Date: Tue, 16 Apr 2024 13:00:06 -0700 Subject: [PATCH] fix linear recomposition Summary: Fixes the torchat ci where we are failing with expand copy. Reviewed By: digantdesai Differential Revision: D56204667 --- backends/xnnpack/passes/convert_to_linear.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backends/xnnpack/passes/convert_to_linear.py b/backends/xnnpack/passes/convert_to_linear.py index 942c09c8ae8..69f882523c8 100644 --- 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