diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp index 394a16f7a..15b5744c6 100644 --- a/tests/test-backend-ops.cpp +++ b/tests/test-backend-ops.cpp @@ -3440,9 +3440,9 @@ static std::vector> make_test_cases_eval() { test_cases.emplace_back(new test_conv_transpose_1d({3,2,1,1}, {3,1,2,1}, 1, 0, 1)); test_cases.emplace_back(new test_conv_transpose_1d({2,1,1,1}, {3,1,1,1}, 1, 0, 1)); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {10, 100 , 1, 1})); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {12, 1024, 1, 1})); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {3 , 5438, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, { 100, 10, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {1024, 12, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {5438, 3, 1, 1})); test_cases.emplace_back(new test_count_equal()); @@ -3824,9 +3824,9 @@ static std::vector> make_test_cases_perf() { test_cases.emplace_back(new test_bin_bcast(ggml_add, GGML_TYPE_F32, {4096, 1, 1, 1}, {1, 1, 1, 1})); test_cases.emplace_back(new test_bin_bcast(ggml_add, GGML_TYPE_F32, {4096, 1, 1, 1}, {1, 512, 1, 1})); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {10, 100 , 1, 1})); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {12, 1024, 1, 1})); - test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {3 , 5438, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, { 100, 10, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {1024, 12, 1, 1})); + test_cases.emplace_back(new test_argmax(GGML_TYPE_F32, {5438, 3, 1, 1})); test_cases.emplace_back(new test_cpy(GGML_TYPE_F32, GGML_TYPE_F16, {512, 3072, 1, 1}));