test-backend-ops : add one more sum_rows test

YellowRoseCx · Dec 12, 2023 · ea4402b · ea4402b
1 parent a51bc0c
commit ea4402b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp
@@ -1372,7 +1372,8 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op
         test_cases.emplace_back(new test_argsort(GGML_TYPE_F32, {16, 10, 10, 10}, order));
     }
 
-    test_cases.emplace_back(new test_sum_rows());
+    test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, {10, 10, 10, 10}));
+    test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, {2, 1, 1, 1}));
 
 #if !defined(__SANITIZE_THREAD__)
     // FIXME: these tests use too much memory with thread sanitizer