Skip to content

Commit 2074498

Browse files
committed
tests: Add im2col perf tests
1 parent e34c5af commit 2074498

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3945,6 +3945,18 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
39453945
}
39463946
}
39473947

3948+
for (int K : {3, 5}) {
3949+
for (int IC : {256, 2560}) {
3950+
for (int IW_IH : {32, 64, 256}) {
3951+
if (IC == 2560 && IW_IH == 256) {
3952+
// too big
3953+
continue;
3954+
}
3955+
test_cases.emplace_back(new test_im2col(GGML_TYPE_F32, GGML_TYPE_F16, GGML_TYPE_F32, {IW_IH, IW_IH, IC, 1}, {K, K, IC, 1}, 1, 1, 1, 1, 1, 1, true));
3956+
}
3957+
}
3958+
}
3959+
39483960
return test_cases;
39493961
}
39503962

0 commit comments

Comments
 (0)