kernels/moe test pruning (#27053)

Signed-off-by: Fardin Hoque <kfhfar@amazon.com> Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>
2025-10-29 21:10:34 -07:00
parent 17d055f527
commit b8c48c5d72
13 changed files with 34 additions and 56 deletions
--- a/tests/kernels/moe/test_cutlass_moe.py
+++ b/tests/kernels/moe/test_cutlass_moe.py
@@ -26,16 +26,13 @@ TOP_KS = [6, 8]

 MNK_FACTORS = [
    (2, 1024, 1024),
-    (2, 1024, 1536),
    (2, 3072, 1024),
    (2, 3072, 1536),
    (7, 3072, 1536),
    (64, 1024, 1024),
    (64, 1024, 1536),
    (64, 3072, 1024),
-    (64, 3072, 1536),
    (224, 1024, 1024),
-    (224, 1024, 1536),
    (224, 3072, 1024),
    (224, 3072, 1536),
    (32768, 1024, 1024),