OLD | NEW |
(Empty) | |
| 1 //===- subzero/src/IceGlobalContext.cpp - Global context defs ---*- C++ -*-===// |
| 2 // |
| 3 // The Subzero Code Generator |
| 4 // |
| 5 // This file is distributed under the University of Illinois Open Source |
| 6 // License. See LICENSE.TXT for details. |
| 7 // |
| 8 //===----------------------------------------------------------------------===// |
| 9 // |
| 10 // This file defines aspects of the compilation that persist across |
| 11 // multiple functions. |
| 12 // |
| 13 //===----------------------------------------------------------------------===// |
| 14 |
| 15 #include "IceDefs.h" |
| 16 #include "IceTypes.h" |
| 17 #include "IceCfg.h" |
| 18 #include "IceGlobalContext.h" |
| 19 #include "IceOperand.h" |
| 20 |
| 21 namespace Ice { |
| 22 |
| 23 // TypePool maps constants of type KeyType (e.g. float) to pointers to |
| 24 // type ValueType (e.g. ConstantFloat). KeyType values are compared |
| 25 // using memcmp() because of potential NaN values in KeyType values. |
| 26 // KeyTypeHasFP indicates whether KeyType is a floating-point type |
| 27 // whose values need to be compared using memcmp() for NaN |
| 28 // correctness. TODO: use std::is_floating_point<KeyType> instead of |
| 29 // KeyTypeHasFP with C++11. |
| 30 template <typename KeyType, typename ValueType, bool KeyTypeHasFP = false> |
| 31 class TypePool { |
| 32 TypePool(const TypePool &) LLVM_DELETED_FUNCTION; |
| 33 TypePool &operator=(const TypePool &) LLVM_DELETED_FUNCTION; |
| 34 |
| 35 public: |
| 36 TypePool() {} |
| 37 ValueType *getOrAdd(GlobalContext *Ctx, Type Ty, KeyType Key) { |
| 38 TupleType TupleKey = std::make_pair(Ty, Key); |
| 39 typename ContainerType::const_iterator Iter = Pool.find(TupleKey); |
| 40 if (Iter != Pool.end()) |
| 41 return Iter->second; |
| 42 ValueType *Result = ValueType::create(Ctx, Ty, Key); |
| 43 Pool[TupleKey] = Result; |
| 44 return Result; |
| 45 } |
| 46 |
| 47 private: |
| 48 typedef std::pair<Type, KeyType> TupleType; |
| 49 struct TupleCompare { |
| 50 bool operator()(const TupleType &A, const TupleType &B) { |
| 51 if (A.first != B.first) |
| 52 return A.first < B.first; |
| 53 if (KeyTypeHasFP) |
| 54 return memcmp(&A.second, &B.second, sizeof(KeyType)) < 0; |
| 55 return A.second < B.second; |
| 56 } |
| 57 }; |
| 58 typedef std::map<const TupleType, ValueType *, TupleCompare> ContainerType; |
| 59 ContainerType Pool; |
| 60 }; |
| 61 |
| 62 // The global constant pool bundles individual pools of each type of |
| 63 // interest. |
| 64 class ConstantPool { |
| 65 ConstantPool(const ConstantPool &) LLVM_DELETED_FUNCTION; |
| 66 ConstantPool &operator=(const ConstantPool &) LLVM_DELETED_FUNCTION; |
| 67 |
| 68 public: |
| 69 ConstantPool() {} |
| 70 TypePool<float, ConstantFloat, true> Floats; |
| 71 TypePool<double, ConstantDouble, true> Doubles; |
| 72 TypePool<uint64_t, ConstantInteger> Integers; |
| 73 TypePool<RelocatableTuple, ConstantRelocatable> Relocatables; |
| 74 }; |
| 75 |
| 76 GlobalContext::GlobalContext(llvm::raw_ostream *OsDump, |
| 77 llvm::raw_ostream *OsEmit, VerboseMask Mask, |
| 78 IceString TestPrefix) |
| 79 : StrDump(OsDump), StrEmit(OsEmit), VMask(Mask), |
| 80 ConstPool(new ConstantPool()), TestPrefix(TestPrefix) {} |
| 81 |
| 82 // In this context, name mangling means to rewrite a symbol using a |
| 83 // given prefix. For a C++ symbol, nest the original symbol inside |
| 84 // the "prefix" namespace. For other symbols, just prepend the |
| 85 // prefix. |
| 86 IceString GlobalContext::mangleName(const IceString &Name) const { |
| 87 // TODO: Add explicit tests (beyond the implicit tests in the linker |
| 88 // that come from the cross tests). |
| 89 // |
| 90 // An already-nested name like foo::bar() gets pushed down one |
| 91 // level, making it equivalent to Prefix::foo::bar(). |
| 92 // _ZN3foo3barExyz ==> _ZN6Prefix3foo3barExyz |
| 93 // A non-nested but mangled name like bar() gets nested, making it |
| 94 // equivalent to Prefix::bar(). |
| 95 // _Z3barxyz ==> ZN6Prefix3barExyz |
| 96 // An unmangled, extern "C" style name, gets a simple prefix: |
| 97 // bar ==> Prefixbar |
| 98 if (getTestPrefix().empty()) |
| 99 return Name; |
| 100 |
| 101 unsigned PrefixLength = getTestPrefix().length(); |
| 102 char NameBase[1 + Name.length()]; |
| 103 const size_t BufLen = 30 + Name.length() + getTestPrefix().length(); |
| 104 char NewName[BufLen]; |
| 105 uint32_t BaseLength = 0; |
| 106 |
| 107 int ItemsParsed = sscanf(Name.c_str(), "_ZN%s", NameBase); |
| 108 if (ItemsParsed == 1) { |
| 109 // Transform _ZN3foo3barExyz ==> _ZN6Prefix3foo3barExyz |
| 110 // (splice in "6Prefix") ^^^^^^^ |
| 111 snprintf(NewName, BufLen, "_ZN%u%s%s", PrefixLength, |
| 112 getTestPrefix().c_str(), NameBase); |
| 113 // We ignore the snprintf return value (here and below). If we |
| 114 // somehow miscalculated the output buffer length, the output will |
| 115 // be truncated, but it will be truncated consistently for all |
| 116 // mangleName() calls on the same input string. |
| 117 return NewName; |
| 118 } |
| 119 |
| 120 ItemsParsed = sscanf(Name.c_str(), "_Z%u%s", &BaseLength, NameBase); |
| 121 if (ItemsParsed == 2) { |
| 122 // Transform _Z3barxyz ==> ZN6Prefix3barExyz |
| 123 // ^^^^^^^^ ^ |
| 124 // (splice in "N6Prefix", and insert "E" after "3bar") |
| 125 char OrigName[Name.length()]; |
| 126 char OrigSuffix[Name.length()]; |
| 127 strncpy(OrigName, NameBase, BaseLength); |
| 128 OrigName[BaseLength] = '\0'; |
| 129 strcpy(OrigSuffix, NameBase + BaseLength); |
| 130 snprintf(NewName, BufLen, "_ZN%u%s%u%sE%s", PrefixLength, |
| 131 getTestPrefix().c_str(), BaseLength, OrigName, OrigSuffix); |
| 132 return NewName; |
| 133 } |
| 134 |
| 135 // Transform bar ==> Prefixbar |
| 136 // ^^^^^^ |
| 137 return getTestPrefix() + Name; |
| 138 } |
| 139 |
| 140 GlobalContext::~GlobalContext() {} |
| 141 |
| 142 Constant *GlobalContext::getConstantInt(Type Ty, uint64_t ConstantInt64) { |
| 143 return ConstPool->Integers.getOrAdd(this, Ty, ConstantInt64); |
| 144 } |
| 145 |
| 146 Constant *GlobalContext::getConstantFloat(float ConstantFloat) { |
| 147 return ConstPool->Floats.getOrAdd(this, IceType_f32, ConstantFloat); |
| 148 } |
| 149 |
| 150 Constant *GlobalContext::getConstantDouble(double ConstantDouble) { |
| 151 return ConstPool->Doubles.getOrAdd(this, IceType_f64, ConstantDouble); |
| 152 } |
| 153 |
| 154 Constant *GlobalContext::getConstantSym(Type Ty, int64_t Offset, |
| 155 const IceString &Name, |
| 156 bool SuppressMangling) { |
| 157 return ConstPool->Relocatables.getOrAdd( |
| 158 this, Ty, RelocatableTuple(Offset, Name, SuppressMangling)); |
| 159 } |
| 160 |
| 161 void Timer::printElapsedUs(GlobalContext *Ctx, const IceString &Tag) const { |
| 162 if (Ctx->isVerbose(IceV_Timing)) { |
| 163 // Prefixing with '#' allows timing strings to be included |
| 164 // without error in textual assembly output. |
| 165 Ctx->getStrDump() << "# " << getElapsedUs() << " usec " << Tag << "\n"; |
| 166 } |
| 167 } |
| 168 |
| 169 } // end of namespace Ice |
OLD | NEW |