Address some build issues

microsoft · Jan 21, 2025 · 5e92e90 · 5e92e90
1 parent c8304fc
commit 5e92e90
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/onnxruntime/core/providers/cuda/cuda_execution_provider.cc b/onnxruntime/core/providers/cuda/cuda_execution_provider.cc
@@ -2735,11 +2735,11 @@ CUDAExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph,
     // Previously assigned nodes have been accounted before
     if (previously_assigned_nodes.count(node_index) > 0 || resource_accountant == nullptr) {
       // XXX: For demo only
-      constexpr const size_t kNodeCountThreshold = 800;
-      static std::atomic_size_t nodes_assigned = 0;
-      if (nodes_assigned.fetch_add(1) > kNodeCountThreshold) {
-        ORT_THROW("CUDA EP is running out of memory");
-      }
+      //constexpr const size_t kNodeCountThreshold = 800;
+      //static std::atomic_size_t nodes_assigned = 0;
+      //if (nodes_assigned.fetch_add(1) > kNodeCountThreshold) {
+      //  ORT_THROW("CUDA EP is running out of memory");
+      //}
       /// XXX: End of DEMO
       auto sub_graph = IndexedSubGraph::Create();
       sub_graph->Nodes().push_back(node_index);

diff --git a/onnxruntime/core/providers/nnapi/nnapi_builtin/nnapi_execution_provider.cc b/onnxruntime/core/providers/nnapi/nnapi_builtin/nnapi_execution_provider.cc
@@ -83,6 +83,7 @@ NnapiExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph_view
                                       const IKernelLookup& /*kernel_lookup*/,
                                       IResourceAccountant* /* resource_accountant */) const {
   std::vector<std::unique_ptr<ComputeCapability>> result;
+  const logging::Logger& logger = *GetLogger();
 
   // TODO: Task 812756: NNAPI EP, add support for subgraph (If and Loop operators)
   if (graph_viewer.IsSubgraph()) {