update fusion_conformer_attention.py

microsoft · Jan 26, 2024 · d61119c · d61119c
1 parent fc44f96
commit d61119c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/python/tools/transformers/fusion_conformer_attention.py b/onnxruntime/python/tools/transformers/fusion_conformer_attention.py
@@ -69,7 +69,7 @@ def fuse(self, normalize_node, input_name_to_nodes, output_name_to_node):
 
         q_nodes = self.model.match_parent_path(
             matmul_qk,
-            ["Div", "Transpose", "Reshape", "Add", "MatMul"],
+            ["Mul", "Transpose", "Reshape", "Add", "MatMul"],
             [0, 0, 0, 0, 1],
         )
         if q_nodes is not None: