fix missing SchemaRegistry when create EPContext model

mingyue · mingyue · commit babf513e4938 · 2024-12-12T03:14:34.000-07:00
diff --git a/onnxruntime/core/framework/graph_partitioner.cc b/onnxruntime/core/framework/graph_partitioner.cc
@@ -681,7 +681,7 @@ static Status CreateEpContextModel(const ExecutionProviders& execution_providers
                            context_cache_path, "' exist already.");
   }
 
-  Model ep_context_model(graph.Name(), false, graph.GetModel().MetaData(), PathString(), IOnnxRuntimeOpSchemaRegistryList(),
+  Model ep_context_model(graph.Name(), false, graph.GetModel().MetaData(), PathString(), IOnnxRuntimeOpSchemaRegistryList{graph.GetSchemaRegistry()},
                          graph.DomainToVersionMap(), {}, logger);
   auto& ep_graph = ep_context_model.MainGraph();
   ep_graph.SetDescription(graph.Description());

Original file line number	Diff line number	Diff line change
`@@ -681,7 +681,7 @@ static Status CreateEpContextModel(const ExecutionProviders& execution_providers`
`681`	`681`	`context_cache_path, "' exist already.");`
`682`	`682`	`}`
`683`	`683`
`684`		`- Model ep_context_model(graph.Name(), false, graph.GetModel().MetaData(), PathString(), IOnnxRuntimeOpSchemaRegistryList(),`
	`684`	`+ Model ep_context_model(graph.Name(), false, graph.GetModel().MetaData(), PathString(), IOnnxRuntimeOpSchemaRegistryList{graph.GetSchemaRegistry()},`
`685`	`685`	`graph.DomainToVersionMap(), {}, logger);`
`686`	`686`	`auto& ep_graph = ep_context_model.MainGraph();`
`687`	`687`	`ep_graph.SetDescription(graph.Description());`