From 703becd796ca7359b6424f74e1884e6de1780dd4 Mon Sep 17 00:00:00 2001
From: Ye Wang <52801275+wangyems@users.noreply.github.com>
Date: Tue, 14 Dec 2021 12:50:35 -0800
Subject: [PATCH] Fix a bug in fusion_embedlayer.py (#10022)

---
 onnxruntime/python/tools/transformers/fusion_embedlayer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/onnxruntime/python/tools/transformers/fusion_embedlayer.py b/onnxruntime/python/tools/transformers/fusion_embedlayer.py
index ccf6f426f1..25cc133755 100644
--- a/onnxruntime/python/tools/transformers/fusion_embedlayer.py
+++ b/onnxruntime/python/tools/transformers/fusion_embedlayer.py
@@ -116,7 +116,7 @@ class FusionEmbedLayerNoMask(Fusion):
             if path1 is None:
                 return False
 
-        expand, shape = path1
+        expand, shape = path1[0], path1[-1]
         if shape.input[0] != input_ids:
             return False