From 32ee379f50721f86d7ad4fdc93d30e277a6d02b5 Mon Sep 17 00:00:00 2001 From: Jeff Bloomfield <38966965+jeffbloo@users.noreply.github.com> Date: Mon, 10 Jan 2022 12:01:08 -0800 Subject: [PATCH] GraphPartitioner.cpp: fixed Merge function to handle duplicated partitions. (#9929) Co-authored-by: Achira --- .../providers/dml/DmlExecutionProvider/src/GraphPartitioner.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/GraphPartitioner.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/GraphPartitioner.cpp index 594f821aaf..0862b07d6b 100644 --- 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; }