From 630dc315ee362a2f7190f695e1b35a25b5d7ff6a Mon Sep 17 00:00:00 2001
From: goostavz <109190422+goostavz@users.noreply.github.com>
Date: Mon, 28 Nov 2022 11:23:20 +0800
Subject: [PATCH] [Triton-MLIR] uncomment the UT in test_gemm that has already
 been fixed (#920)

---
 python/tests/test_gemm.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/tests/test_gemm.py b/python/tests/test_gemm.py
index b12bd6fad..0edd53b30 100644
--- a/python/tests/test_gemm.py
+++ b/python/tests/test_gemm.py
@@ -187,7 +187,8 @@ def get_proper_err(a, b, golden):
     [128, 256, 128, 4, 128, 256, 32, False, False],
     [256, 128, 64, 4, 256, 128, 16, False, False],
     [128, 64, 128, 4, 128, 64, 32, False, False],
-    # [16, 16, 64, 4, 16, 16, 16, False, False],  # TODO failed due to pipeline pass
+    [16, 16, 64, 4, 16, 16, 16, False, False],
+    [32, 32, 64, 4, 32, 32, 32, False, False],
     # trans
     [128, 64, 128, 4, 128, 64, 32, True, False],
     [128, 64, 128, 4, 128, 64, 32, False, True],