graph: dnnl: catch sdpa creation error for fallback

Author: xiang1guo

src/graph/backend/dnnl/op_executable.hpp

@@ -2685,9 +2685,12 @@ struct sdpa_executable_t : public op_executable_t {
 
         dim_t kv_head_number
                 = op->get_input_value(1)->get_logical_tensor().dims[1];
-        create_sdpa_pd(sdpa_pd_, p_engine.get(), md_q.get(), md_k.get(),
-                md_v.get(), md_dst.get(), md_mask.get(), scale_dt,
+        status_t s = create_sdpa_pd(sdpa_pd_, p_engine.get(), md_q.get(),
+                md_k.get(), md_v.get(), md_dst.get(), md_mask.get(), scale_dt,
                 is_invert_scale_, kv_head_number, is_causal_mask_, attr.get());
+        if (s != dnnl::impl::status::success) {
+            throw std::runtime_error("create_sdpa_pd failed");
+        }
 
         sdpa_pd_->create_primitive(sdpa_prim_, p_engine.get());
     }

src/graph/backend/dnnl/passes/compile_ops.cpp

@@ -59,14 +59,20 @@ status_t compile_ops(std::shared_ptr<subgraph_t> &sg) {
         auto cur_op = op->shared_from_this();
         auto creator = opm->get_additional_item<executable_creator_func>(
                 "executable_creator");
-        std::shared_ptr<op_executable_t> exec
-                = creator(cur_op, p_engine, mgr, pd_cache);
 
-        VCHECK_COMPILE_OPS(exec != nullptr, status::invalid_graph_op,
-                "unimplemented op, can't compile op %s",
-                op->get_name().c_str());
+        try {
+            std::shared_ptr<op_executable_t> exec
+                    = creator(cur_op, p_engine, mgr, pd_cache);
+            VCHECK_COMPILE_OPS(exec != nullptr, status::invalid_graph_op,
+                    "unimplemented op, can't compile op %s",
+                    op->get_name().c_str());
 
-        sg->execs_.emplace_back(exec);
+            sg->execs_.emplace_back(exec);
+        } catch (const std::runtime_error &e) {
+            VCHECK_COMPILE_OPS(false, status::unimplemented,
+                    "failed to create executable for op %s: %s",
+                    op->get_name().c_str(), e.what());
+        }
         sg->is_constant_.push_back(op->has_attr(op_attr::is_constant)
                 && op->get_attr<bool>(op_attr::is_constant));
         return status::success;