diff --git a/lib/Transforms/IPO/MergeFunctions.cpp b/lib/Transforms/IPO/MergeFunctions.cpp index 559ef0ba228..2fb0ddb174a 100644 --- a/lib/Transforms/IPO/MergeFunctions.cpp +++ b/lib/Transforms/IPO/MergeFunctions.cpp @@ -766,13 +766,23 @@ int FunctionComparator::cmpOperation(const Instruction *L, if (int Res = cmpNumbers(CI->getCallingConv(), cast(R)->getCallingConv())) return Res; - return cmpAttrs(CI->getAttributes(), cast(R)->getAttributes()); + if (int Res = + cmpAttrs(CI->getAttributes(), cast(R)->getAttributes())) + return Res; + return cmpNumbers( + (uint64_t)CI->getMetadata(LLVMContext::MD_range), + (uint64_t)cast(R)->getMetadata(LLVMContext::MD_range)); } if (const InvokeInst *CI = dyn_cast(L)) { if (int Res = cmpNumbers(CI->getCallingConv(), cast(R)->getCallingConv())) return Res; - return cmpAttrs(CI->getAttributes(), cast(R)->getAttributes()); + if (int Res = + cmpAttrs(CI->getAttributes(), cast(R)->getAttributes())) + return Res; + return cmpNumbers( + (uint64_t)CI->getMetadata(LLVMContext::MD_range), + (uint64_t)cast(R)->getMetadata(LLVMContext::MD_range)); } if (const InsertValueInst *IVI = dyn_cast(L)) { ArrayRef LIndices = IVI->getIndices(); diff --git a/test/Transforms/MergeFunc/call-and-invoke-with-ranges.ll b/test/Transforms/MergeFunc/call-and-invoke-with-ranges.ll new file mode 100644 index 00000000000..9878b471500 --- /dev/null +++ b/test/Transforms/MergeFunc/call-and-invoke-with-ranges.ll @@ -0,0 +1,91 @@ +; RUN: opt -mergefunc -S < %s | FileCheck %s + +define i8 @call_with_range() { + bitcast i8 0 to i8 ; dummy to make the function large enough + %out = call i8 @dummy(), !range !0 + ret i8 %out +} + +define i8 @call_no_range() { +; CHECK-LABEL: @call_no_range +; CHECK-NEXT: bitcast i8 0 to i8 +; CHECK-NEXT: %out = call i8 @dummy() +; CHECK-NEXT: ret i8 %out + bitcast i8 0 to i8 + %out = call i8 @dummy() + ret i8 %out +} + +define i8 @call_different_range() { +; CHECK-LABEL: @call_different_range +; CHECK-NEXT: bitcast i8 0 to i8 +; CHECK-NEXT: %out = call i8 @dummy(), !range !1 +; CHECK-NEXT: ret i8 %out + bitcast i8 0 to i8 + %out = call i8 @dummy(), !range !1 + ret i8 %out +} + +define i8 @invoke_with_range() { + %out = invoke i8 @dummy() to label %next unwind label %lpad, !range !0 + +next: + ret i8 %out + +lpad: + %pad = landingpad { i8*, i32 } personality i8* undef cleanup + resume { i8*, i32 } zeroinitializer +} + +define i8 @invoke_no_range() { +; CHECK-LABEL: @invoke_no_range() +; CHECK-NEXT: invoke i8 @dummy + %out = invoke i8 @dummy() to label %next unwind label %lpad + +next: + ret i8 %out + +lpad: + %pad = landingpad { i8*, i32 } personality i8* undef cleanup + resume { i8*, i32 } zeroinitializer +} + +define i8 @invoke_different_range() { +; CHECK-LABEL: @invoke_different_range() +; CHECK-NEXT: invoke i8 @dummy + %out = invoke i8 @dummy() to label %next unwind label %lpad, !range !1 + +next: + ret i8 %out + +lpad: + %pad = landingpad { i8*, i32 } personality i8* undef cleanup + resume { i8*, i32 } zeroinitializer +} + +define i8 @call_same_range() { +; CHECK-LABEL: @call_same_range +; CHECK: tail call i8 @call_with_range + bitcast i8 0 to i8 + %out = call i8 @dummy(), !range !0 + ret i8 %out +} + +define i8 @invoke_same_range() { +; CHECK-LABEL: @invoke_same_range() +; CHECK: tail call i8 @invoke_with_range() + %out = invoke i8 @dummy() to label %next unwind label %lpad, !range !0 + +next: + ret i8 %out + +lpad: + %pad = landingpad { i8*, i32 } personality i8* undef cleanup + resume { i8*, i32 } zeroinitializer +} + +declare i8 @dummy(); +declare i32 @__gxx_personality_v0(...) + +!0 = metadata !{i8 0, i8 2} +!1 = metadata !{i8 5, i8 7} \ No newline at end of file