From ea4402bb0e2e70737545b6869f9eff995b988a36 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Tue, 12 Dec 2023 17:03:38 +0200 Subject: [PATCH] test-backend-ops : add one more sum_rows test --- tests/test-backend-ops.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp index 8f4b470bc..44830b4d4 100644 --- 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