[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:
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
|
||||
|
||||
Reference in new issue
Block a user