rm logs

zz002 · Aug 13, 2024 · 80d07f6 · 80d07f6
1 parent 1038920
commit 80d07f6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git 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<NodeIndex> 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()<<" "<<node->GetExecutionProviderType()<< std::endl;
-                                    return kernel_def == nullptr; }),
+                                   [&](NodeIndex index) { return kernel_lookup.LookUpKernel(*graph.GetNode(index)) == nullptr; }),
                     node_indexs.end());
   std::vector<std::unique_ptr<ComputeCapability>> 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
@@ -217,7 +217,6 @@ void create_kernel_registry(std::vector<OrtCustomOpDomain*> 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<OrtCustomOpDomain*> 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 {