Skip to content

Commit

Permalink
[GPU] Fix empty oneDnn users (openvinotoolkit#26900)
Browse files Browse the repository at this point in the history
### Details:
 - Fix empty oneDnn users
  • Loading branch information
p-durandin authored Oct 3, 2024
1 parent 0ae3683 commit 890f2e1
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions src/plugins/intel_gpu/src/graph/layout_optimizer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1089,11 +1089,13 @@ format layout_optimizer::get_expected_format(quantize_node const& node) {
auto use_onednn_impls = _optimization_attributes.use_onednn_impls;

if (use_onednn_impls) {
auto& user = node.get_users().front();
if (user != nullptr && user->get_preferred_input_fmt(user->get_dependency_index(node)) != format::any) {
expected = user->get_preferred_input_fmt(user->get_dependency_index(node));
} else {
expected = format::any;
expected = format::any;
auto& users = node.get_users();
if (users.size() != 0) {
auto& user = users.front();
if (user != nullptr && user->get_preferred_input_fmt(user->get_dependency_index(node)) != format::any) {
expected = user->get_preferred_input_fmt(user->get_dependency_index(node));
}
}
} else if (only_gemm_users(node)) {
// TODO: Gemm is not supporting fsv layouts
Expand Down

0 comments on commit 890f2e1

Please sign in to comment.