Skip to content

Commit 8c7cb62

Browse files
Add more unittests
1 parent 4fc7d18 commit 8c7cb62

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

src/plugins/intel_gpu/tests/unit/test_cases/pooling_gpu_test.cpp

+34
Original file line numberDiff line numberDiff line change
@@ -2333,6 +2333,40 @@ class pooling_scale_random_test_base : public pooling_random_test_base<InputT, M
23332333
VF<output_t> _shift;
23342334
};
23352335

2336+
using pooling_random_test_int8_uint8 = pooling_random_test;
2337+
2338+
TEST_P(pooling_random_test_int8_uint8, avg_int8) {
2339+
auto test_case = pooling_random_test_base<int8_t, pooling_mode::average>();
2340+
ASSERT_NO_FATAL_FAILURE(test_case.run_random(GetParam(), false));
2341+
}
2342+
2343+
TEST_P(pooling_random_test_int8_uint8, max_int8) {
2344+
auto test_case = pooling_random_test_base<int8_t, pooling_mode::max>();
2345+
ASSERT_NO_FATAL_FAILURE(test_case.run_random(GetParam(), false));
2346+
}
2347+
2348+
TEST_P(pooling_random_test_int8_uint8, avg_uint8) {
2349+
auto test_case = pooling_random_test_base<uint8_t, pooling_mode::average>();
2350+
ASSERT_NO_FATAL_FAILURE(test_case.run_random(GetParam(), false));
2351+
}
2352+
2353+
TEST_P(pooling_random_test_int8_uint8, max_uint8) {
2354+
auto test_case = pooling_random_test_base<uint8_t, pooling_mode::max>();
2355+
ASSERT_NO_FATAL_FAILURE(test_case.run_random(GetParam(), false));
2356+
}
2357+
2358+
INSTANTIATE_TEST_SUITE_P(
2359+
smoke_low_precision,
2360+
pooling_random_test_int8_uint8,
2361+
testing::Combine(testing::Values(1, 2),
2362+
testing::Values(3, 8),
2363+
testing::Values(std::tuple<size_t, size_t, size_t>(12, 12, 1), std::tuple<size_t, size_t, size_t>(24, 24, 1)),
2364+
testing::Values(std::tuple<size_t, size_t, size_t>(4, 4, 1), std::tuple<size_t, size_t, size_t>(2, 2, 1)),
2365+
testing::Values(std::tuple<int, int, int>(2, 2, 1)),
2366+
testing::Values(std::tuple<int, int, int>(0, 0, 0)),
2367+
testing::Values(format::fs_b_yx_fsv32)),
2368+
testing::internal::DefaultParamName<pooling_random_test_params>);
2369+
23362370
using pooling_random_test_fp16_fp32 = pooling_random_test;
23372371

23382372
TEST_P(pooling_random_test_fp16_fp32, avg_fp16) {

0 commit comments

Comments
 (0)