diff --git a/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc b/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc index a2dcafbbaa..8afe83098c 100644 --- a/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc +++ b/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc @@ -92,6 +92,10 @@ std::shared_ptr VitisAIExecutionProvider::GetKernelRegistry() co std::vector> VitisAIExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph, const IKernelLookup& /*kernel_lookup*/) const { + if (graph.IsSubgraph()) { + // VITIS AI EP not support sungraph. Assigned to CPU. + return {}; + } auto opt_str = info_.get_json_config_str(); // String execution_providers_ = std::make_unique(compile_onnx_model(graph, *GetLogger(), opt_str));