@@ -482,6 +482,13 @@ class gemm_2in_dynamic_add : public gemm_2in_add {};
482
482
TEST_P (gemm_2in_dynamic_add, add) {
483
483
auto p = GetParam ();
484
484
485
+ if (engine.get_device_info ().supports_immad ) {
486
+ p.expected_fused_primitives ++;
487
+ if (!p.kernel_name .empty ()) {
488
+ p.expected_not_fused_primitives ++;
489
+ }
490
+ }
491
+
485
492
cfg_fused.set_property (ov::intel_gpu::allow_new_shape_infer (true ));
486
493
cfg_not_fused.set_property (ov::intel_gpu::allow_new_shape_infer (true ));
487
494
@@ -518,10 +525,13 @@ TEST_P(gemm_2in_dynamic_add, add) {
518
525
}
519
526
520
527
INSTANTIATE_TEST_SUITE_P (fusings_gpu, gemm_2in_dynamic_add, ::testing::ValuesIn(std::vector<gemm_test_params>{
521
- gemm_test_params{ CASE_GEMM_2IN_FP16_3D_1, 4 , 4 , " gemm_tiled_opt" , broadcast_kinds::batch, eltwise_mode::sum },
528
+ gemm_test_params{ CASE_GEMM_2IN_FP16_3D_1, 4 , 5 , " " , broadcast_kinds::batch, eltwise_mode::sum },
529
+ gemm_test_params{ CASE_GEMM_2IN_FP16_3D_1, 4 , 5 , " " , broadcast_kinds::feature, eltwise_mode::sum },
530
+ gemm_test_params{ CASE_GEMM_2IN_FP16_3D_2, 4 , 5 , " " , broadcast_kinds::feature, eltwise_mode::sum },
531
+ gemm_test_params{ CASE_GEMM_2IN_FP16_3D_2, 4 , 5 , " " , broadcast_kinds::batch, eltwise_mode::sum },
532
+ // Reference graph can be fused because force_implementation leads to optimize_data = true;
522
533
gemm_test_params{ CASE_GEMM_2IN_FP16_3D_1, 4 , 4 , " gemm_tiled_opt" , broadcast_kinds::feature, eltwise_mode::sum },
523
- gemm_test_params{ CASE_GEMM_2IN_FP16_3D_2, 4 , 4 , " gemm_tiled_opt" , broadcast_kinds::feature, eltwise_mode::sum },
524
- gemm_test_params{ CASE_GEMM_2IN_FP16_3D_2, 4 , 4 , " gemm_tiled_opt" , broadcast_kinds::batch, eltwise_mode::sum },
534
+ gemm_test_params{ CASE_GEMM_2IN_FP16_3D_1, 4 , 4 , " gemm_tiled_opt" , broadcast_kinds::batch, eltwise_mode::sum }
525
535
}));
526
536
527
537
class gemm_2in_act_scale_quantize_i8 : public GemmFusingTest {};
0 commit comments