Merge branch 'user/anagrao/CPConstCPUData' into user/anagrao/CPDedupl…

…icate
microsoft · Nov 29, 2023 · 465361e · 465361e
2 parents 790093a + 95d8a3a
commit 465361e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp
@@ -842,7 +842,11 @@ namespace Dml
         {
             return;
         }
-
+        else if (!constExpTensor->IsCpuData())
+        {
+            return;
+        }
+
         uint32_t totalKernelInputElementCount = constExpTensor->GetTotalElementCount();
         if (totalKernelInputElementCount <= 1)
         {