GraphPartitioner.cpp: fixed Merge function to handle duplicated parti…

…tions. (#9929) Co-authored-by: Achira <[email protected]>
microsoft · Jan 10, 2022 · 32ee379 · 32ee379
1 parent 5cd57bb
commit 32ee379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/GraphPartitioner.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/GraphPartitioner.cpp
@@ -105,7 +105,7 @@ namespace Dml
 
         for (GraphPartition* partitionToMerge : partitionsToMerge)
         {
-            if (partitionToMerge == this)
+            if (partitionToMerge->GetRootMergedPartition() == this)
             {
                 continue;
             }