Fix ifdef

Try inlined hashset
microsoft · Jan 26, 2024 · 11f3c5d · 11f3c5d
1 parent 5268ee5
commit 11f3c5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/coreml/builders/impl/conv_op_builder.cc b/onnxruntime/core/providers/coreml/builders/impl/conv_op_builder.cc
@@ -11,10 +11,10 @@
 #include "core/providers/coreml/builders/impl/builder_utils.h"
 #include "core/providers/coreml/builders/model_builder.h"
 #include "core/providers/coreml/shape_utils.h"
-#endif
 
 using namespace CoreML::Specification;
 using namespace CoreML::Specification::MILSpec;
+#endif
 
 namespace onnxruntime {
 namespace coreml {

diff --git a/onnxruntime/core/providers/coreml/coreml_execution_provider.h b/onnxruntime/core/providers/coreml/coreml_execution_provider.h
@@ -3,6 +3,7 @@
 
 #pragma once
 
+#include "core/common/inlined_containers.h"
 #include "core/framework/execution_provider.h"
 #include "core/providers/coreml/coreml_provider_factory.h"
 
@@ -33,7 +34,7 @@ class CoreMLExecutionProvider : public IExecutionProvider {
   const int32_t coreml_version_;
   // <fused_node_name, <coreml_model_file_path, compiled_coreml_model>>
 #ifdef __APPLE__OR__TEST__
-  std::unordered_map<std::string, std::unique_ptr<onnxruntime::coreml::Model>> coreml_models_;
+  InlinedHashMap<std::string, std::unique_ptr<onnxruntime::coreml::Model>> coreml_models_;
 #endif
 };
 }  // namespace onnxruntime