minor fix

microsoft · Nov 1, 2024 · 96b7f1d · 96b7f1d
1 parent 4b48a38
commit 96b7f1d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/onnxruntime/test/util/default_providers.cc b/onnxruntime/test/util/default_providers.cc
@@ -259,8 +259,9 @@ std::unique_ptr<IExecutionProvider> DefaultCoreMLExecutionProvider(bool use_mlpr
   if (use_mlprogram) {
     coreml_flags |= COREML_FLAG_CREATE_MLPROGRAM;
   }
-
-  return CoreMLProviderFactoryCreator::Create(coreml_flags)->CreateProvider();
+  auto option = ProviderOptions();
+  option["coreml_flags"] = coreml_flags;
+  return CoreMLProviderFactoryCreator::Create(option)->CreateProvider();
 #else
   ORT_UNUSED_PARAMETER(use_mlprogram);
   return nullptr;