Teach MergeFunctions about operand bundles

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@255528 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Sanjoy Das 2015-12-14 19:11:40 +00:00
parent 97a58d8cf4
commit 32d33b59bc
2 changed files with 65 additions and 0 deletions

View File

@ -407,6 +407,7 @@ private:
int cmpMem(StringRef L, StringRef R) const;
int cmpAttrs(const AttributeSet L, const AttributeSet R) const;
int cmpRangeMetadata(const MDNode* L, const MDNode* R) const;
int cmpOperandBundlesSchema(const Instruction *L, const Instruction *R) const;
// The two functions undergoing comparison.
const Function *FnL, *FnR;
@ -564,6 +565,32 @@ int FunctionComparator::cmpRangeMetadata(const MDNode* L,
return 0;
}
int FunctionComparator::cmpOperandBundlesSchema(const Instruction *L,
const Instruction *R) const {
ImmutableCallSite LCS(L);
ImmutableCallSite RCS(R);
assert(LCS && RCS && "Must be calls or invokes!");
assert(LCS.isCall() == RCS.isCall() && "Can't compare otherwise!");
if (int Res =
cmpNumbers(LCS.getNumOperandBundles(), RCS.getNumOperandBundles()))
return Res;
for (unsigned i = 0, e = LCS.getNumOperandBundles(); i != e; ++i) {
auto OBL = LCS.getOperandBundleAt(i);
auto OBR = RCS.getOperandBundleAt(i);
if (int Res = OBL.getTagName().compare(OBR.getTagName()))
return Res;
if (int Res = cmpNumbers(OBL.Inputs.size(), OBR.Inputs.size()))
return Res;
}
return 0;
}
/// Constants comparison:
/// 1. Check whether type of L constant could be losslessly bitcasted to R
/// type.
@ -941,6 +968,8 @@ int FunctionComparator::cmpOperations(const Instruction *L,
if (int Res =
cmpAttrs(CI->getAttributes(), cast<CallInst>(R)->getAttributes()))
return Res;
if (int Res = cmpOperandBundlesSchema(CI, R))
return Res;
return cmpRangeMetadata(
CI->getMetadata(LLVMContext::MD_range),
cast<CallInst>(R)->getMetadata(LLVMContext::MD_range));
@ -952,6 +981,8 @@ int FunctionComparator::cmpOperations(const Instruction *L,
if (int Res =
cmpAttrs(CI->getAttributes(), cast<InvokeInst>(R)->getAttributes()))
return Res;
if (int Res = cmpOperandBundlesSchema(CI, R))
return Res;
return cmpRangeMetadata(
CI->getMetadata(LLVMContext::MD_range),
cast<InvokeInst>(R)->getMetadata(LLVMContext::MD_range));

View File

@ -0,0 +1,34 @@
; RUN: opt -S -mergefunc < %s | FileCheck %s
; Minor note: functions need to be at least three instructions long
; to be considered by -mergefunc.
declare i32 @foo(...)
define i32 @f() {
; CHECK-LABEL: @f(
entry:
%v0 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
%v1 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
%v2 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
; CHECK: %v0 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
; CHECK: %v1 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
; CHECK: %v2 = call i32 (...) @foo(i32 10) [ "foo"(i32 20) ]
ret i32 %v2
}
define i32 @g() {
; CHECK-LABEL: @g(
entry:
%v0 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
%v1 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
%v2 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
; CHECK: %v0 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
; CHECK: %v1 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
; CHECK: %v2 = call i32 (...) @foo() [ "foo"(i32 10, i32 20) ]
ret i32 %v2
}