IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Add additional handling of fused activations for OneDNN primitives (#8004)

Browse Source
This commit is contained in:
Sergey Shlyapnikov 2021-10-15 09:31:49 +03:00 committed by GitHub
parent 897cd09a5a
commit 8ea9986896
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
inference-engine/thirdparty/clDNN/src/impls/onednn/primitive_onednn_base.h vendored
View File

@ -783,6 +783,17 @@ protected:
} }
} }
if (cldnn_post_ops.size() && arg.get_fused_activations_funcs().size())
throw std::runtime_error("Unsupported mix of fused ops and activations");
for (size_t i = 0; i < arg.get_fused_activations_funcs().size(); i++) {
auto activation_type = arg.get_fused_activations_funcs()[i];
auto params = arg.get_fused_activations_params()[i];
dnnl::algorithm alg = onednn::convert_activation_func(activation_type);
post_ops.append_eltwise(1.0f, alg, params.a, params.b);
update_onednn_post_op_list(onednn_post_op_type::eltwise_act, empty_mem);
}
// Update total onednn post-ops info // Update total onednn post-ops info
onednn_fusing_map.emplace(arg.id(), std::move(fused_ops)); onednn_fusing_map.emplace(arg.id(), std::move(fused_ops));