cpu: x64: jit_reorder: direct_copy: restrict plain-strided cases

dzarukin · vpirogov · commit a762d3248ee5 · 2025-05-22T16:10:14.000-07:00
diff --git a/src/cpu/x64/jit_uni_reorder_direct_copy.cpp b/src/cpu/x64/jit_uni_reorder_direct_copy.cpp
@@ -311,6 +311,13 @@ status_t jit_uni_reorder_direct_copy_t::pd_t::init(
     VDISPATCH_REORDER(src_d.similar_to(dst_d, true, false, 0),
             VERBOSE_TENSOR_FORMAT_MISMATCH, "src", "dst");
 
+    // Direct copy doesn't operate on plain non-dense cases, though it can
+    // simply extend nelems to contain wholes in strides but it would be
+    // inefficient.
+    // TODO: enable smart plain non-dense case support.
+    VDISPATCH_REORDER(IMPLICATION(src_d.is_plain(), src_d.is_dense()),
+            VERBOSE_UNSUPPORTED_TENSOR_LAYOUT, "src");
+
     VDISPATCH_REORDER(src_d.extra().flags == dst_d.extra().flags,
             VERBOSE_UNSUPPORTED_MD_FLAG, "src or dst");
 
