Skip to content

Commit e8373bf

Browse files
committed
Add unit-test for CUB-based mean
Tests should run with CUDA Graphs enabled per default on NVGPUs
1 parent 48cf9e4 commit e8373bf

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6006,6 +6006,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
60066006
test_cases.emplace_back(new test_sum(GGML_TYPE_F32, { 33, 256, 1, 1 }));
60076007
test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, { 33, 256, 1, 1 }));
60086008
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, { 33, 256, 1, 1 }));
6009+
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, { 32769, 1, 1, 1 }));
60096010
test_cases.emplace_back(new test_group_norm(GGML_TYPE_F32, {64, 64, 320, 1}));
60106011
test_cases.emplace_back(new test_group_norm(GGML_TYPE_F32, {9, 9, 1280, 1}));
60116012
test_cases.emplace_back(new test_acc());

0 commit comments

Comments
 (0)