[FRONTEND] Added a few assertions in semantic.dot (#977)

2022-12-12 00:07:14 -08:00
parent e552219104
commit e5cfa0f633
2 changed files with 6 additions and 1 deletions
--- a/python/tests/test_gemm.py
+++ b/python/tests/test_gemm.py
@@ -72,7 +72,7 @@ def test_gemm_no_scf(SHAPE, NUM_WARPS, TRANS_A, TRANS_B):
    for shape in [
        [64, 128, 128],
        [128, 128, 128],
-        [16, 8, 32],
+        [16, 16, 32],
        [32, 16, 64],
        [32, 16, 64],
    ]