Temporary change for constant graph node enumeration

microsoft · Oct 2, 2023 · 660a407 · 660a407
1 parent 80479dd
commit 660a407
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/DmlGraphFusionHelper.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/DmlGraphFusionHelper.cpp
@@ -283,11 +283,6 @@ namespace DmlGraphFusionHelper
         DML_PREVIEW_OPERATOR_FIRST = 0xC0000000,
     };
 
-    enum DML_GRAPH_NODE_TYPE_PREVIEW
-    { 
-        DML_GRAPH_NODE_TYPE_CONSTANT_DATA = 0xCC000000,  
-    }; 
-
     struct DML_CONSTANT_DATA_GRAPH_NODE_DESC_PREVIEW 
     { 
         const BYTE* data;
@@ -324,7 +319,8 @@ namespace DmlGraphFusionHelper
                 nodeInfo.name.data()
                 };
 
-                dmlGraphNodes[i] = DML_GRAPH_NODE_DESC{(DML_GRAPH_NODE_TYPE) DML_GRAPH_NODE_TYPE_CONSTANT_DATA, &dmlConstantGraphNodes[i]};
+                // TODO: Change as new header is ingested
+                dmlGraphNodes[i] = DML_GRAPH_NODE_DESC{(DML_GRAPH_NODE_TYPE) 2, &dmlConstantGraphNodes[i]};
             }
         }