[GPU] Fix issues with floating point fusings support for cldnn / onednn fully connected kernels (#10519)

* [GPU] Fix of floating point fusings inside fc kernels

* [GPU] Fix for related tests
This commit is contained in:
Ilya Znamenskiy authored and GitHub committed 2022-02-21 12:03:23 +09:00
1 parent a52c755d21
commit 7c93902dac
2 files changed
+5 -9

No files matched your search

@@ -554,13 +554,9 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {
};
auto fc_supports_fusings = [](fully_connected_node& node) -> bool {
auto out_fmt = node.get_output_layout().format;
auto in_l = node.get_dependency(0).get_output_layout();
auto in_dt = in_l.data_type;
auto in_b = in_l.size.batch[0];
auto in_dt = node.get_dependency(0).get_output_layout().data_type;
return (data_type_traits::is_i8_u8(in_dt) || (data_type_traits::is_floating_point(in_dt) && in_b > 1)) &&
out_fmt != format::yxfb;
return data_type_traits::is_i8_u8(in_dt);
};
auto gemm_supports_fusings = [](gemm_node& node) -> bool {
@@ -359,9 +359,9 @@ INSTANTIATE_TEST_SUITE_P(fusings_gpu, fc_int8_scale_activation_quantize_i8, ::te
fully_connected_test_params{ CASE_FC_U8S8_3D_2, 2, 5 },
fully_connected_test_params{ CASE_FC_U8S8_3D_3, 2, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_1, 2, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_2, 2, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_3, 2, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_1, 3, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_2, 3, 5 },
fully_connected_test_params{ CASE_FC_FP32_3D_3, 3, 5 },
}));
#ifdef ENABLE_ONEDNN_FOR_GPU