test-backend-ops : add one more sum_rows test

ggerganov · ggerganov · commit ea4402bb0e2e · 2023-12-12T17:03:38.000+02:00
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

Original file line number	Diff line number	Diff line change
`@@ -1372,7 +1372,8 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op`
`1372`	`1372`	`test_cases.emplace_back(new test_argsort(GGML_TYPE_F32, {16, 10, 10, 10}, order));`
`1373`	`1373`	`}`
`1374`	`1374`
`1375`		`- test_cases.emplace_back(new test_sum_rows());`
	`1375`	`+ test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, {10, 10, 10, 10}));`
	`1376`	`+ test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, {2, 1, 1, 1}));`
`1376`	`1377`
`1377`	`1378`	`#if !defined(__SANITIZE_THREAD__)`
`1378`	`1379`	`// FIXME: these tests use too much memory with thread sanitizer`