@@ -158,15 +158,14 @@ TEST_P(mvn_scale_quantize_i8, basic) {
158
158
}
159
159
160
160
INSTANTIATE_TEST_SUITE_P (fusings_gpu, mvn_scale_quantize_i8, ::testing::ValuesIn(std::vector<mvn_test_params>{
161
- // Full fusing for fp input not supported yet, it may lead to output padding and non-optimal kernel
162
- // mvn_test_params{ CASE_MVN_F32_1, 2, 4 },
163
- // mvn_test_params{ CASE_MVN_F32_2, 2, 4 },
164
- // mvn_test_params{ CASE_MVN_3D_F32_1, 2, 4 },
165
- // mvn_test_params{ CASE_MVN_3D_F32_2, 2, 4 },
166
- // mvn_test_params{ CASE_MVN_F16_1, 2, 4 },
167
- // mvn_test_params{ CASE_MVN_F16_2, 2, 4 },
168
- // mvn_test_params{ CASE_MVN_3D_F16_1, 2, 4 },
169
- // mvn_test_params{ CASE_MVN_3D_F16_2, 2, 4 },
161
+ mvn_test_params{ CASE_MVN_F32_1, 2 , 2 , 4 },
162
+ mvn_test_params{ CASE_MVN_F32_2, 2 , 2 , 4 },
163
+ mvn_test_params{ CASE_MVN_3D_F32_1, 2 , 2 , 4 },
164
+ mvn_test_params{ CASE_MVN_3D_F32_2, 2 , 2 , 4 },
165
+ mvn_test_params{ CASE_MVN_F16_1, 2 , 2 , 4 },
166
+ mvn_test_params{ CASE_MVN_F16_2, 2 , 2 , 4 },
167
+ mvn_test_params{ CASE_MVN_3D_F16_1, 2 , 2 , 4 },
168
+ mvn_test_params{ CASE_MVN_3D_F16_2, 2 , 2 , 4 },
170
169
mvn_test_params{ CASE_MVN_I8_1, 2 , 2 , 4 },
171
170
mvn_test_params{ CASE_MVN_I8_2, 2 , 2 , 4 },
172
171
mvn_test_params{ CASE_MVN_I8_3, 2 , 2 , 4 },
@@ -207,15 +206,14 @@ TEST_P(mvn_scale_activation_eltwise_fp32_quantize_i8, basic) {
207
206
}
208
207
209
208
INSTANTIATE_TEST_SUITE_P (fusings_gpu, mvn_scale_activation_eltwise_fp32_quantize_i8, ::testing::ValuesIn(std::vector<mvn_test_params>{
210
- // Full using for fp input not supported yet, it may lead to output padding and non-optimal kernel
211
- // mvn_test_params{ CASE_MVN_F32_1, 2, 7 },
212
- // mvn_test_params{ CASE_MVN_F32_2, 2, 7 },
213
- // mvn_test_params{ CASE_MVN_3D_F32_1, 2, 7 },
214
- // mvn_test_params{ CASE_MVN_3D_F32_2, 2, 7 },
215
- // mvn_test_params{ CASE_MVN_F16_1, 2, 7 },
216
- // mvn_test_params{ CASE_MVN_F16_2, 2, 7 },
217
- // mvn_test_params{ CASE_MVN_3D_F16_1, 2, 7 },
218
- // mvn_test_params{ CASE_MVN_3D_F16_2, 2, 7 },
209
+ mvn_test_params{ CASE_MVN_F32_1, 2 , 4 , 6 },
210
+ mvn_test_params{ CASE_MVN_F32_2, 2 , 4 , 6 },
211
+ mvn_test_params{ CASE_MVN_3D_F32_1, 2 , 4 , 6 },
212
+ mvn_test_params{ CASE_MVN_3D_F32_2, 2 , 4 , 6 },
213
+ mvn_test_params{ CASE_MVN_F16_1, 2 , 4 , 6 },
214
+ mvn_test_params{ CASE_MVN_F16_2, 2 , 4 , 6 },
215
+ mvn_test_params{ CASE_MVN_3D_F16_1, 2 , 4 , 6 },
216
+ mvn_test_params{ CASE_MVN_3D_F16_2, 2 , 4 , 6 },
219
217
mvn_test_params{ CASE_MVN_I8_1, 2 , 4 , 6 },
220
218
mvn_test_params{ CASE_MVN_I8_2, 2 , 4 , 6 },
221
219
mvn_test_params{ CASE_MVN_I8_3, 2 , 4 , 6 },
0 commit comments