| 12345678910111213141516171819202122232425262728293031323334 |
- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- #ifndef C10_UTIL_FBCODEMAPS_H_
- #define C10_UTIL_FBCODEMAPS_H_
- // Map typedefs so that we can use folly's F14 maps in fbcode without
- // taking a folly dependency.
- #ifdef FBCODE_CAFFE2
- #include <folly/container/F14Map.h>
- #include <folly/container/F14Set.h>
- #else
- #include <unordered_map>
- #include <unordered_set>
- #endif
- namespace c10 {
- #ifdef FBCODE_CAFFE2
- template <typename Key, typename Value>
- using FastMap = folly::F14FastMap<Key, Value>;
- template <typename Key>
- using FastSet = folly::F14FastSet<Key>;
- #else
- template <typename Key, typename Value>
- using FastMap = std::unordered_map<Key, Value>;
- template <typename Key>
- using FastSet = std::unordered_set<Key>;
- #endif
- } // namespace c10
- #endif // C10_UTIL_FBCODEMAPS_H_
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
