From f41407dfb9a3e297a9b60705836f7061db4a8a02 Mon Sep 17 00:00:00 2001 From: "zaixing.wang" Date: Wed, 9 Oct 2024 15:58:28 +0800 Subject: [PATCH] add ut --- .../intel_gpu/tests/unit/fusions/select_fusion_test.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/plugins/intel_gpu/tests/unit/fusions/select_fusion_test.cpp b/src/plugins/intel_gpu/tests/unit/fusions/select_fusion_test.cpp index 0cc5843545ba44..2751ded346398b 100644 --- a/src/plugins/intel_gpu/tests/unit/fusions/select_fusion_test.cpp +++ b/src/plugins/intel_gpu/tests/unit/fusions/select_fusion_test.cpp @@ -63,9 +63,12 @@ class SelectFusingTest : public ::BaseFusingTest { } // namespace #define CASE_SELECT_FP32_TO_I8_0 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f32, data_types::i8, format::bfyx, format::bfyx +#define CASE_SELECT_FP32_TO_U8_0 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f32, data_types::u8, format::bfyx, format::bfyx #define CASE_SELECT_FP32_TO_F16_0 {2, 16, 17, 4}, {2, 16, 1, 4}, data_types::f32, data_types::f16, format::bfyx, format::bfyx #define CASE_SELECT_FP16_TO_I8_0 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f16, data_types::i8, format::bfyx, format::bfyx +#define CASE_SELECT_FP16_TO_U8_0 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f16, data_types::u8, format::bfyx, format::bfyx #define CASE_SELECT_FP16_TO_I8_1 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f16, data_types::i8, format::bfyx, format::bfzyx +#define CASE_SELECT_FP16_TO_U8_1 {2, 16, 4, 4}, {2, 16, 4, 4}, data_types::f16, data_types::u8, format::bfyx, format::bfzyx class select_reorder_fusion : public SelectFusingTest {}; TEST_P(select_reorder_fusion, basic) { @@ -85,8 +88,11 @@ TEST_P(select_reorder_fusion, basic) { INSTANTIATE_TEST_SUITE_P(fusings_gpu, select_reorder_fusion, ::testing::ValuesIn(std::vector{ select_test_params{ CASE_SELECT_FP32_TO_F16_0, 5, 6}, select_test_params{ CASE_SELECT_FP32_TO_I8_0, 5, 6}, + select_test_params{ CASE_SELECT_FP32_TO_U8_0, 5, 6}, select_test_params{ CASE_SELECT_FP16_TO_I8_0, 5, 6}, + select_test_params{ CASE_SELECT_FP16_TO_U8_0, 5, 6}, select_test_params{ CASE_SELECT_FP16_TO_I8_1, 6, 6}, // reorder should not be fused + select_test_params{ CASE_SELECT_FP16_TO_U8_1, 6, 6}, // reorder should not be fused })); class select_reorder_fusion_dynamic : public SelectFusingTest {};