diff --git a/onnxruntime/core/providers/vitisai/imp/capability.cc b/onnxruntime/core/providers/vitisai/imp/capability.cc index ad33cbaedab18..f2cefe9dff48e 100644 --- a/onnxruntime/core/providers/vitisai/imp/capability.cc +++ b/onnxruntime/core/providers/vitisai/imp/capability.cc @@ -52,15 +52,9 @@ GetComputeCapabilityOps(const onnxruntime::GraphViewer& graph, std::vector node_indexs = graph.GetNodesInTopologicalOrder(); node_indexs.erase(std::remove_if(node_indexs.begin(), node_indexs.end(), [&](NodeIndex index) { return all_nodes_included_eps.count(index) > 0; }), node_indexs.end()); node_indexs.erase(std::remove_if(node_indexs.begin(), node_indexs.end(), - [&](NodeIndex index) { - auto node = graph.GetNode(index); - const KernelCreateInfo* kernel_def = kernel_lookup.LookUpKernel(*node); - std::cout << "index "<< index <<" "<<(const void*) kernel_def <<" "<< node->OpType()<<" " - << node->Name()<<" "<GetExecutionProviderType()<< std::endl; - return kernel_def == nullptr; }), + [&](NodeIndex index) { return kernel_lookup.LookUpKernel(*graph.GetNode(index)) == nullptr; }), node_indexs.end()); std::vector> result; - std::cout << "node_indexs.size() "<< node_indexs.size() << std::endl; for (auto& n : node_indexs) { auto indexed_subgraph = IndexedSubGraph::Create(); indexed_subgraph->Nodes() = {n}; diff --git a/onnxruntime/core/providers/vitisai/imp/global_api.cc b/onnxruntime/core/providers/vitisai/imp/global_api.cc index ebb0a9cb9c96e..9638905f1b10b 100644 --- a/onnxruntime/core/providers/vitisai/imp/global_api.cc +++ b/onnxruntime/core/providers/vitisai/imp/global_api.cc @@ -217,7 +217,6 @@ void create_kernel_registry(std::vector domains) { if ( schema->typeConstraintMap().count(schema->inputs__GetTypeStr(i))){ input_name = schema->inputs__GetTypeStr(i); } - std::cout << input_name << " "<< input_type<<" "<< ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED << std::endl; if (input_type == ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED) { def_builder->TypeConstraint(input_name.c_str(), DataTypeImpl::AllTensorTypes()); } else { @@ -232,7 +231,6 @@ void create_kernel_registry(std::vector domains) { output_name = schema->outputs__GetTypeStr(i); } } - std::cout << output_name << " "<< output_type<<" "<< ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED << std::endl; if (output_type == ONNX_TENSOR_ELEMENT_DATA_TYPE_UNDEFINED) { def_builder->TypeConstraint(output_name.c_str(), DataTypeImpl::AllTensorTypes()); } else {