use byxf format when fsv2 is selected in select_preferred_formats_for…

…_onednn
openvinotoolkit · Aug 2, 2024 · 0a67997 · 0a67997
1 parent c7c8743
commit 0a67997
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/impls/onednn/utils.cpp b/src/plugins/intel_gpu/src/graph/impls/onednn/utils.cpp
@@ -345,12 +345,14 @@ cldnn::format find_data_format(dnnl::memory::desc desc) {
                     break;
                 }
             }
-            // b_fs_yx_fsv2 has dnnl::memory::format_tag::undef , so not select the format.
-            if (static_cast<format::type>(fmt_idx) == format::type::b_fs_yx_fsv2) {
-                is_match = false;
+
+            if (is_match) {
+                // b_fs_yx_fsv2 has dnnl::memory::format_tag::undef , so select byfx instead of b_fs_yx_fsv2.
+                if (static_cast<format::type>(fmt_idx) == format::type::b_fs_yx_fsv2)
+                    return format::type::byxf;
+                else
+                    return static_cast<format::type>(fmt_idx);
             }
-            if (is_match)
-                return static_cast<format::type>(fmt_idx);
         }
     }