diff options
| -rw-r--r-- | CodeGen/CodeGenModule.cpp | 14 | ||||
| -rw-r--r-- | CodeGen/CodeGenModule.h | 2 |
2 files changed, 15 insertions, 1 deletions
diff --git a/CodeGen/CodeGenModule.cpp b/CodeGen/CodeGenModule.cpp index d68fbe616d..e3181337bf 100644 --- a/CodeGen/CodeGenModule.cpp +++ b/CodeGen/CodeGenModule.cpp @@ -31,7 +31,7 @@ CodeGenModule::CodeGenModule(ASTContext &C, const LangOptions &LO, llvm::Module &M, const llvm::TargetData &TD, Diagnostic &diags) : Context(C), Features(LO), TheModule(M), TheTargetData(TD), Diags(diags), - Types(C, M, TD), MemCpyFn(0), CFConstantStringClassRef(0) {} + Types(C, M, TD), MemCpyFn(0), MemSetFn(0), CFConstantStringClassRef(0) {} /// WarnUnsupported - Print out a warning that codegen doesn't support the /// specified stmt yet. @@ -329,6 +329,18 @@ llvm::Function *CodeGenModule::getMemCpyFn() { return MemCpyFn = getIntrinsic(IID); } +llvm::Function *CodeGenModule::getMemSetFn() { + if (MemSetFn) return MemSetFn; + llvm::Intrinsic::ID IID; + uint64_t Size; unsigned Align; + Context.Target.getPointerInfo(Size, Align, FullSourceLoc()); + switch (Size) { + default: assert(0 && "Unknown ptr width"); + case 32: IID = llvm::Intrinsic::memset_i32; break; + case 64: IID = llvm::Intrinsic::memset_i64; break; + } + return MemSetFn = getIntrinsic(IID); +} llvm::Constant *CodeGenModule:: GetAddrOfConstantCFString(const std::string &str) { diff --git a/CodeGen/CodeGenModule.h b/CodeGen/CodeGenModule.h index 92060887af..48283c53a6 100644 --- a/CodeGen/CodeGenModule.h +++ b/CodeGen/CodeGenModule.h @@ -52,6 +52,7 @@ class CodeGenModule { CodeGenTypes Types; llvm::Function *MemCpyFn; + llvm::Function *MemSetFn; llvm::DenseMap<const Decl*, llvm::Constant*> GlobalDeclMap; llvm::StringMap<llvm::Constant*> CFConstantStringMap; @@ -85,6 +86,7 @@ public: /// array containing the literal. The result is pointer to array type. llvm::Constant *GetAddrOfConstantString(const std::string& str); llvm::Function *getMemCpyFn(); + llvm::Function *getMemSetFn(); llvm::Function *getIntrinsic(unsigned IID, const llvm::Type **Tys = 0, unsigned NumTys = 0); |