CUDA: add mean operation (ggml-org#14313)

am17an · Minh141120 · commit c00c96748ef7 · 2025-07-05T23:51:18.000+07:00
* CUDA: add mean operation

* add back sum_rows_f32_cuda

* Review: early exit if col!=0
diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp
@@ -4967,6 +4967,8 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
 
     test_cases.emplace_back(new test_conv_transpose_2d({256, 256, 256, 1}, {3, 3, 16, 256}, 1));
 
+    test_cases.emplace_back(new test_mean(GGML_TYPE_F32, {256, 256, 3, 1}));
+
     return test_cases;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -4967,6 +4967,8 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {`
`4967`	`4967`
`4968`	`4968`	`test_cases.emplace_back(new test_conv_transpose_2d({256, 256, 256, 1}, {3, 3, 16, 256}, 1));`
`4969`	`4969`
	`4970`	`+ test_cases.emplace_back(new test_mean(GGML_TYPE_F32, {256, 256, 3, 1}));`
	`4971`	`+`
`4970`	`4972`	`return test_cases;`
`4971`	`4973`	`}`
`4972`	`4974`