#0: Clean up tests

tenstorrent · Feb 7, 2024 · fb5613e · fb5613e
1 parent 4c78d84
commit fb5613e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/tt_eager/python_api_testing/unit_testing/test_attn_matmul.py b/tests/tt_eager/python_api_testing/unit_testing/test_attn_matmul.py
@@ -110,11 +110,9 @@ def test_attn_matmul_with_program_cache(in0_dtype, in1_dtype, out_dtype, device,
 @pytest.mark.parametrize(
     "batch, K, seq_len, q_heads, kv_heads",
     (
-        (32, 64, 512 + 32, 10, 5),
-        (32, 64, 512 + 64, 16, 1),
         (32, 64, 512 + 96, 32, 2),
-        (32, 64, 2048, 32, 2),
-        # (32, 512 + 64, 64, 32, 2),
+        (32, 1024 + 32, 64, 32, 2),
+        (32, 64, 128, 71, 1),
     ),
 )
 def test_group_attn_matmul(