diff options
| -rw-r--r-- | llvm/lib/Transforms/IPO/HotColdSplitting.cpp | 3 | ||||
| -rw-r--r-- | llvm/test/Transforms/HotColdSplit/retain-section.ll | 33 |
2 files changed, 36 insertions, 0 deletions
diff --git a/llvm/lib/Transforms/IPO/HotColdSplitting.cpp b/llvm/lib/Transforms/IPO/HotColdSplitting.cpp index 9db079c7734..2bd3df3add7 100644 --- a/llvm/lib/Transforms/IPO/HotColdSplitting.cpp +++ b/llvm/lib/Transforms/IPO/HotColdSplitting.cpp @@ -328,6 +328,9 @@ Function *HotColdSplitting::extractColdRegion( } CI->setIsNoInline(); + if (OrigF->hasSection()) + OutF->setSection(OrigF->getSection()); + markFunctionCold(*OutF, BFI != nullptr); LLVM_DEBUG(llvm::dbgs() << "Outlined Region: " << *OutF); diff --git a/llvm/test/Transforms/HotColdSplit/retain-section.ll b/llvm/test/Transforms/HotColdSplit/retain-section.ll new file mode 100644 index 00000000000..8c0cf604962 --- /dev/null +++ b/llvm/test/Transforms/HotColdSplit/retain-section.ll @@ -0,0 +1,33 @@ +; RUN: opt -hotcoldsplit -hotcoldsplit-threshold=0 -S < %s | FileCheck %s + +target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128" +target triple = "x86_64-apple-macosx10.14.0" + +; Retain section after cold code is split off. + +; CHECK-LABEL: @fun +; CHECK: call void @fun.cold.1{{.*}} section ".text.cold" +define void @fun() section ".text.cold" { +entry: + br i1 undef, label %if.then, label %if.else + +if.then: + ret void + +if.else: + call void @sink() + ret void +} + +; CHECK: define {{.*}} @foo{{.*}}#[[outlined_func_attr:[0-9]+]] +define void @foo() cold { + ret void +} + +declare void @sink() cold + +; CHECK: define {{.*}} @fun.cold.1{{.*}}#[[outlined_func_attr]] + +; CHECK: attributes #[[outlined_func_attr]] = { +; CHECK-SAME: cold +; CHECK-SAME: minsize |
