Procházet zdrojové kódy

tests: add performance test for mul mat id (#15543)

Eve před 4 měsíci
rodič
revize
44b1efa41a
1 změnil soubory, kde provedl 18 přidání a 0 odebrání
  1. 18 0
      tests/test-backend-ops.cpp

+ 18 - 0
tests/test-backend-ops.cpp

@@ -6400,6 +6400,24 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
         }
         }
     }
     }
 
 
+    // qwen3-30b-a3b
+    for (int bs : {1, 4, 8, 512}) {
+        for (ggml_type type_a : {GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_Q4_0, GGML_TYPE_Q8_0, GGML_TYPE_Q4_K, GGML_TYPE_Q6_K, GGML_TYPE_IQ2_XS}) {
+            for (ggml_type type_b : {GGML_TYPE_F32}) {
+                test_cases.emplace_back(new test_mul_mat_id(type_a, type_b, 128, 8, false, 768, bs, 2048, 1));
+            }
+        }
+    }
+
+    // gpt-oss-20b
+    for (int bs : {1, 4, 8, 512}) {
+        for (ggml_type type_a : {GGML_TYPE_MXFP4}) {
+            for (ggml_type type_b : {GGML_TYPE_F32}) {
+                test_cases.emplace_back(new test_mul_mat_id(type_a, type_b, 32, 4, false, 2880, bs, 2880, 1));
+            }
+        }
+    }
+
     for (int K : {3, 5}) {
     for (int K : {3, 5}) {
         for (int IC : {256, 2560}) {
         for (int IC : {256, 2560}) {
             for (int IW_IH : {32, 64, 256}) {
             for (int IW_IH : {32, 64, 256}) {