File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
onnxruntime/core/providers/openvino/ov_versions Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -171,19 +171,17 @@ std::vector<std::unique_ptr<ComputeCapability>> GetCapability::Execute() {
171
171
bool omit_subgraph = false ;
172
172
173
173
auto id = this_cluster.at (0 );
174
- std::cout << graph_viewer_.GetNode (id)->Name () << " \n " ;
175
-
176
174
if (this_cluster.size () == 1 ) {
177
175
// check next cluster
178
176
auto index = this_cluster.at (0 );
179
- if (graph_viewer_.GetNode (index)->OpType () != " EPContext" ) {
177
+ if (graph_viewer_.GetNode (index)->OpType () == " EPContext" ) {
178
+ omit_subgraph=false ;
179
+ } else if (cluster_index < this_cluster.size ()-1 ) {
180
180
bool append_node = AddTrivialClusterToNextClusterIfConnected (graph_viewer_, index, connected_clusters[cluster_index+1 ]);
181
181
if (append_node) {
182
182
connected_clusters[cluster_index+1 ].emplace_back (index);
183
183
}
184
184
omit_subgraph=true ;
185
- } else if (graph_viewer_.GetNode (index)->OpType () == " EPContext" ) {
186
- omit_subgraph=false ;
187
185
}
188
186
}
189
187
You can’t perform that action at this time.
0 commit comments