graph: backend: dnnl: add check for sdp decomp kernel

filter out implicit causal mask in sdp decomp kernel
oneapi-src · Jan 21, 2025 · 749c60b · 749c60b
1 parent 2dc60a7
commit 749c60b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/graph/backend/dnnl/kernels/sdp_decomp_config.cpp b/src/graph/backend/dnnl/kernels/sdp_decomp_config.cpp
@@ -457,6 +457,8 @@ impl::status_t sdp_decomp_config_t::record_input_offset(
                     graph::op_kind::SoftMax};
     for (const auto &cur_op : sg->get_ops()) {
         const auto &op_kind = cur_op->get_kind();
+        VCHECK_SDP_DECOMP(op_kind == graph::op_kind::GenIndex,
+                status::unimplemented, "Not support implicit causal mask");
         VCHECK_SDP_DECOMP(
                 !(op_kind == graph::op_kind::DynamicDequantize
                         && cur_op->get_attr<std::string>(op_attr::qtype)