Implement CodeGenModule::getMemSetFn method.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@47346 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/CodeGen/CodeGenModule.cpp b/CodeGen/CodeGenModule.cpp
index d68fbe6..e318133 100644
--- a/CodeGen/CodeGenModule.cpp
+++ b/CodeGen/CodeGenModule.cpp
@@ -31,7 +31,7 @@
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 @@
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 9206088..48283c5 100644
--- a/CodeGen/CodeGenModule.h
+++ b/CodeGen/CodeGenModule.h
@@ -52,6 +52,7 @@
CodeGenTypes Types;
llvm::Function *MemCpyFn;
+ llvm::Function *MemSetFn;
llvm::DenseMap<const Decl*, llvm::Constant*> GlobalDeclMap;
llvm::StringMap<llvm::Constant*> CFConstantStringMap;
@@ -85,6 +86,7 @@
/// 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);