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],