update oneDNN to 9ef80d173 on main (#2370)

* update oneDNN to 9ef80d173 on main * update softmax UT due to backend change
intel · Dec 18, 2023 · 1364dc5 · 1364dc5
1 parent 760d204
commit 1364dc5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/tests/cpu/test_softmax.py b/tests/cpu/test_softmax.py
@@ -62,6 +62,13 @@ def forward(self, x):
         attention_scores = torch.matmul(x, x.transpose(-1, -2))
         attention_scores = attention_scores / 64
         attention_scores = nn.Softmax(dim=-1)(attention_scores)
+        # Previously, the backend requires the pattern to be dequant-matmul-div-add-softmax
+        # so that they could be selected into the MHA partition.
+        # Now, dequant-matmul-div-softmax can hit the MHA pattern since add
+        # becomes optional in the pattern matcher.
+        # We add a softmax call here to test the functionality of ipex::softmax_ when
+        # it's input is an oneDNN graph fusion group.
+        attention_scores = nn.Softmax(dim=-1)(attention_scores)
         return attention_scores
 
 

diff --git a/third_party/ideep b/third_party/ideep