update

microsoft · Oct 5, 2023 · 409170b · 409170b
1 parent d5c3452
commit 409170b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/onnxruntime/core/providers/js/js_execution_provider.cc b/onnxruntime/core/providers/js/js_execution_provider.cc
@@ -649,6 +649,10 @@ std::vector<std::unique_ptr<ComputeCapability>> JsExecutionProvider::GetCapabili
 
     const auto& node = *p_node;
     if (!node.GetExecutionProviderType().empty()) {
+	  // If the node was added by layout transformer, do not move it to CPU
+	  if(node.GetExecutionProviderType() == kJsExecutionProvider) {
+		 candidates.push_back(node.Index());
+	  }
       continue;
     }
 
@@ -659,17 +663,14 @@ std::vector<std::unique_ptr<ComputeCapability>> JsExecutionProvider::GetCapabili
       continue;
     }
     candidates.push_back(node.Index());
-    // If the node was added by layout transformer, do not move it to CPU
-    if(node.Domain() != "com.ms.internal.nhwc"){ 
-	  LOGS(*GetLogger(), ERROR) << "Add node to  tenative_candidates. node domain: " << node.Domain() << " node name: " << node.Name();
-      tenative_candidates.push_back(node.Index());
-    }
+    tenative_candidates.push_back(node.Index());
   }
   auto cpu_nodes = GetCpuPreferredNodes(graph, kernel_lookup, tenative_candidates);
   std::vector<std::unique_ptr<ComputeCapability>> result;
   for (auto& node_index : candidates) {
-    if (cpu_nodes.count(node_index) > 0)
+    if (cpu_nodes.count(node_index) > 0) {
       continue;
+	}
 
     auto sub_graph = std::make_unique<IndexedSubGraph>();
     sub_graph->nodes.push_back(node_index);