From 87285323e683e7adc48cf3eab17a00ca0b4fe8e2 Mon Sep 17 00:00:00 2001 From: BoarQing Date: Fri, 11 Aug 2023 00:45:13 +0800 Subject: [PATCH] [VITISAI] nested subgraph is unsupported for now (#17067) ### Description return empty ComputeCapability when a graph contains nested subgraph. ### Motivation and Context For now, our architecture does not support nested subgraph. So, we return empty ComputeCapability for this case. --- .../core/providers/vitisai/vitisai_execution_provider.cc | 4 ++++ 1 file changed, 4 insertions(+) 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));