diff --git a/lib/Transforms/Scalar/DeadStoreElimination.cpp b/lib/Transforms/Scalar/DeadStoreElimination.cpp index c8c53606015..a46e802f4c2 100644 --- a/lib/Transforms/Scalar/DeadStoreElimination.cpp +++ b/lib/Transforms/Scalar/DeadStoreElimination.cpp @@ -282,6 +282,12 @@ static uint64_t getPointerSize(const Value *V, AliasAnalysis &AA) { return C->getZExtValue(); } + if (const CallInst *CI = extractCallocCall(V)) { + if (const ConstantInt *C1 = dyn_cast(CI->getArgOperand(0))) + if (const ConstantInt *C2 = dyn_cast(CI->getArgOperand(1))) + return (C1->getValue() * C2->getValue()).getZExtValue(); + } + if (TD == 0) return AliasAnalysis::UnknownSize; @@ -704,9 +710,11 @@ bool DSE::handleEndBlock(BasicBlock &BB) { // Okay, so these are dead heap objects, but if the pointer never escapes // then it's leaked by this function anyways. - if (CallInst *CI = extractMallocCall(I)) - if (!PointerMayBeCaptured(CI, true, true)) - DeadStackObjects.insert(CI); + CallInst *CI = extractMallocCall(I); + if (!CI) + CI = extractCallocCall(I); + if (CI && !PointerMayBeCaptured(CI, true, true)) + DeadStackObjects.insert(CI); } // Treat byval arguments the same, stores to them are dead at the end of the @@ -759,6 +767,11 @@ bool DSE::handleEndBlock(BasicBlock &BB) { continue; } + if (CallInst *CI = extractCallocCall(BBI)) { + DeadStackObjects.erase(CI); + continue; + } + if (CallSite CS = cast(BBI)) { // If this call does not access memory, it can't be loading any of our // pointers. diff --git a/lib/Transforms/Utils/Local.cpp b/lib/Transforms/Utils/Local.cpp index d1c4d596823..4e1a00fa2d3 100644 --- a/lib/Transforms/Utils/Local.cpp +++ b/lib/Transforms/Utils/Local.cpp @@ -260,7 +260,7 @@ bool llvm::isInstructionTriviallyDead(Instruction *I) { return isa(II->getArgOperand(1)); } - if (extractMallocCall(I)) return true; + if (extractMallocCall(I) || extractCallocCall(I)) return true; if (CallInst *CI = isFreeCall(I)) if (Constant *C = dyn_cast(CI->getArgOperand(0))) diff --git a/test/Transforms/DeadStoreElimination/simple.ll b/test/Transforms/DeadStoreElimination/simple.ll index 81eb5a8c705..7c8a9b3337e 100644 --- a/test/Transforms/DeadStoreElimination/simple.ll +++ b/test/Transforms/DeadStoreElimination/simple.ll @@ -164,7 +164,7 @@ define i32* @test13() { } declare noalias i8* @malloc(i32) - +declare noalias i8* @calloc(i32, i32) define void @test14(i32* %Q) { @@ -258,3 +258,11 @@ define void @test20() { } ; CHECK: @test20 ; CHECK-NEXT: ret void + +; CHECK: @test21 +define void @test21() { + %m = call i8* @calloc(i32 9, i32 7) + store i8 0, i8* %m +; CHECK-NEXT: ret void + ret void +}