[][std::unordered_map] Optimize c10::Registry::Create (#6609)

Reviewed By: bikramSingh91 Differential Revision: D49355735 fbshipit-source-id: 3d78819b909b83d5985851b601e1ba9747a800b4
facebookincubator · Dec 2, 2023 · ea01c43 · ea01c43
1 parent 525fc01
commit ea01c43
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/velox/common/serialization/Registry.h b/velox/common/serialization/Registry.h
@@ -28,6 +28,7 @@
 
 #include <glog/logging.h>
 #include "folly/Preprocessor.h"
+#include "folly/container/F14Map.h"
 
 #include "velox/common/base/Exceptions.h"
 #include "velox/core/Metaprogramming.h"
@@ -47,7 +48,7 @@ template <class KeyType, class FunctionSignature>
 class Registry {
  public:
   using Creator = std::function<FunctionSignature>;
-  using CreatorMap = std::unordered_map<KeyType, Creator>;
+  using CreatorMap = folly::F14NodeMap<KeyType, Creator>;
 
   Registry() : Create(creatorMap_) {}