x64: lrn, softmax: move setting formats before post_ops_ok check

oneapi-src · Apr 11, 2024 · 5806bab · 5806bab
1 parent f47e44c
commit 5806bab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/cpu/x64/jit_uni_layer_normalization.cpp b/src/cpu/x64/jit_uni_layer_normalization.cpp
@@ -1233,9 +1233,9 @@ status_t jit_uni_layer_normalization_fwd_t::pd_t::init(engine_t *engine) {
 
         return injector::post_ops_ok(post_ops_args);
     };
-    VDISPATCH_LNORM(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
     VDISPATCH_LNORM(attr_.set_default_formats(dst_md(0)) == status::success,
             VERBOSE_UNSUPPORTED_POSTOP);
+    VDISPATCH_LNORM(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
 
     VDISPATCH_LNORM(fill_compatible_stats_md(*src_md(), reordered_stat_md_)
                     == status::success,

diff --git a/src/cpu/x64/jit_uni_softmax.hpp b/src/cpu/x64/jit_uni_softmax.hpp
@@ -135,13 +135,13 @@ struct jit_uni_softmax_fwd_t : public primitive_t {
                             | skip_mask_t::post_ops),
                     VERBOSE_UNSUPPORTED_ATTR);
             VDISPATCH_SOFTMAX(attr_scales_ok(), VERBOSE_UNSUPPORTED_SCALES_CFG);
-            VDISPATCH_SOFTMAX(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
 
             VDISPATCH_SOFTMAX(set_default_formats() == status::success,
                     VERBOSE_UNSUPPORTED_TAG);
             VDISPATCH_SOFTMAX(
                     attr_.set_default_formats(dst_md(0)) == status::success,
                     VERBOSE_UNSUPPORTED_TAG);
+            VDISPATCH_SOFTMAX(post_ops_ok(), VERBOSE_UNSUPPORTED_POSTOP);
             VDISPATCH_SOFTMAX(
                     memory_desc_wrapper(src_md()).similar_to(
                             memory_desc_wrapper(dst_md()), true, false, 0),