[llvm] 4d7a0ab - [DataLayout] Change return type of `getStackAlignment` to `MaybeAlign` (#105478)
via llvm-commits
llvm-commits at lists.llvm.org
Tue Aug 27 12:59:37 PDT 2024
Author: Sergei Barannikov
Date: 2024-08-27T22:59:33+03:00
New Revision: 4d7a0abae8b40482a44c22257b2e86c6fa89547c
URL: https://github.com/llvm/llvm-project/commit/4d7a0abae8b40482a44c22257b2e86c6fa89547c
DIFF: https://github.com/llvm/llvm-project/commit/4d7a0abae8b40482a44c22257b2e86c6fa89547c.diff
LOG: [DataLayout] Change return type of `getStackAlignment` to `MaybeAlign` (#105478)
Currently, `getStackAlignment` asserts if the stack alignment wasn't
specified. This makes it inconvenient to use and complicates testing.
This change also makes `exceedsNaturalStackAlignment` method redundant.
Added:
Modified:
llvm/include/llvm/IR/DataLayout.h
llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
llvm/lib/Target/AArch64/AArch64CallingConvention.cpp
llvm/lib/Target/ARM/ARMCallingConv.cpp
llvm/lib/Target/ARM/ARMISelLowering.cpp
llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
llvm/lib/Transforms/IPO/ExpandVariadics.cpp
llvm/lib/Transforms/Utils/Local.cpp
llvm/unittests/IR/DataLayoutTest.cpp
Removed:
################################################################################
diff --git a/llvm/include/llvm/IR/DataLayout.h b/llvm/include/llvm/IR/DataLayout.h
index 2f06bda6c30a51..145f1a29c7dfb7 100644
--- a/llvm/include/llvm/IR/DataLayout.h
+++ b/llvm/include/llvm/IR/DataLayout.h
@@ -220,15 +220,9 @@ class DataLayout {
bool isIllegalInteger(uint64_t Width) const { return !isLegalInteger(Width); }
- /// Returns true if the given alignment exceeds the natural stack alignment.
- bool exceedsNaturalStackAlignment(Align Alignment) const {
- return StackNaturalAlign && (Alignment > *StackNaturalAlign);
- }
-
- Align getStackAlignment() const {
- assert(StackNaturalAlign && "StackNaturalAlign must be defined");
- return *StackNaturalAlign;
- }
+ /// Returns the natural stack alignment, or MaybeAlign() if one wasn't
+ /// specified.
+ MaybeAlign getStackAlignment() const { return StackNaturalAlign; }
unsigned getAllocaAddrSpace() const { return AllocaAddrSpace; }
diff --git a/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp b/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
index 3fece81df1f2fd..f818c4e1426ce1 100644
--- a/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
+++ b/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
@@ -9136,8 +9136,8 @@ LegalizerHelper::lowerMemcpy(MachineInstr &MI, Register Dst, Register Src,
// realignment.
const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
if (!TRI->hasStackRealignment(MF))
- while (NewAlign > Alignment && DL.exceedsNaturalStackAlignment(NewAlign))
- NewAlign = NewAlign.previous();
+ if (MaybeAlign StackAlign = DL.getStackAlignment())
+ NewAlign = std::min(NewAlign, *StackAlign);
if (NewAlign > Alignment) {
Alignment = NewAlign;
@@ -9244,8 +9244,8 @@ LegalizerHelper::lowerMemmove(MachineInstr &MI, Register Dst, Register Src,
// realignment.
const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
if (!TRI->hasStackRealignment(MF))
- while (NewAlign > Alignment && DL.exceedsNaturalStackAlignment(NewAlign))
- NewAlign = NewAlign.previous();
+ if (MaybeAlign StackAlign = DL.getStackAlignment())
+ NewAlign = std::min(NewAlign, *StackAlign);
if (NewAlign > Alignment) {
Alignment = NewAlign;
diff --git a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
index 27675dce70c260..9c7a43064ecf62 100644
--- a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
@@ -7896,8 +7896,8 @@ static SDValue getMemcpyLoadsAndStores(SelectionDAG &DAG, const SDLoc &dl,
// optimization.
const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
if (!TRI->hasStackRealignment(MF))
- while (NewAlign > Alignment && DL.exceedsNaturalStackAlignment(NewAlign))
- NewAlign = NewAlign.previous();
+ if (MaybeAlign StackAlign = DL.getStackAlignment())
+ NewAlign = std::min(NewAlign, *StackAlign);
if (NewAlign > Alignment) {
// Give the stack frame object a larger alignment if needed.
@@ -8091,8 +8091,8 @@ static SDValue getMemmoveLoadsAndStores(SelectionDAG &DAG, const SDLoc &dl,
// optimization.
const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
if (!TRI->hasStackRealignment(MF))
- while (NewAlign > Alignment && DL.exceedsNaturalStackAlignment(NewAlign))
- NewAlign = NewAlign.previous();
+ if (MaybeAlign StackAlign = DL.getStackAlignment())
+ NewAlign = std::min(NewAlign, *StackAlign);
if (NewAlign > Alignment) {
// Give the stack frame object a larger alignment if needed.
@@ -8209,8 +8209,8 @@ static SDValue getMemsetStores(SelectionDAG &DAG, const SDLoc &dl,
// optimization.
const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
if (!TRI->hasStackRealignment(MF))
- while (NewAlign > Alignment && DL.exceedsNaturalStackAlignment(NewAlign))
- NewAlign = NewAlign.previous();
+ if (MaybeAlign StackAlign = DL.getStackAlignment())
+ NewAlign = std::min(NewAlign, *StackAlign);
if (NewAlign > Alignment) {
// Give the stack frame object a larger alignment if needed.
diff --git a/llvm/lib/Target/AArch64/AArch64CallingConvention.cpp b/llvm/lib/Target/AArch64/AArch64CallingConvention.cpp
index 9a804c12939c4b..1eb34e7ca4ddc6 100644
--- a/llvm/lib/Target/AArch64/AArch64CallingConvention.cpp
+++ b/llvm/lib/Target/AArch64/AArch64CallingConvention.cpp
@@ -209,10 +209,11 @@ static bool CC_AArch64_Custom_Block(unsigned &ValNo, MVT &ValVT, MVT &LocVT,
State.AllocateReg(Reg);
}
- const Align StackAlign =
+ const MaybeAlign StackAlign =
State.getMachineFunction().getDataLayout().getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
const Align MemAlign = ArgFlags.getNonZeroMemAlign();
- Align SlotAlign = std::min(MemAlign, StackAlign);
+ Align SlotAlign = std::min(MemAlign, *StackAlign);
if (!Subtarget.isTargetDarwin())
SlotAlign = std::max(SlotAlign, Align(8));
diff --git a/llvm/lib/Target/ARM/ARMCallingConv.cpp b/llvm/lib/Target/ARM/ARMCallingConv.cpp
index 2ab66da4b4d2d9..a8333ffd210d01 100644
--- a/llvm/lib/Target/ARM/ARMCallingConv.cpp
+++ b/llvm/lib/Target/ARM/ARMCallingConv.cpp
@@ -190,9 +190,10 @@ static bool CC_ARM_AAPCS_Custom_Aggregate(unsigned ValNo, MVT ValVT,
// Try to allocate a contiguous block of registers, each of the correct
// size to hold one member.
auto &DL = State.getMachineFunction().getDataLayout();
- const Align StackAlign = DL.getStackAlignment();
+ const MaybeAlign StackAlign = DL.getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
const Align FirstMemberAlign(PendingMembers[0].getExtraInfo());
- Align Alignment = std::min(FirstMemberAlign, StackAlign);
+ Align Alignment = std::min(FirstMemberAlign, *StackAlign);
ArrayRef<MCPhysReg> RegList;
switch (LocVT.SimpleTy) {
diff --git a/llvm/lib/Target/ARM/ARMISelLowering.cpp b/llvm/lib/Target/ARM/ARMISelLowering.cpp
index 853f54943eebf1..ec6367a803506b 100644
--- a/llvm/lib/Target/ARM/ARMISelLowering.cpp
+++ b/llvm/lib/Target/ARM/ARMISelLowering.cpp
@@ -2461,8 +2461,9 @@ ARMTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI,
// Since callee will pop argument stack as a tail call, we must keep the
// popped size 16-byte aligned.
- Align StackAlign = DAG.getDataLayout().getStackAlignment();
- NumBytes = alignTo(NumBytes, StackAlign);
+ MaybeAlign StackAlign = DAG.getDataLayout().getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
+ NumBytes = alignTo(NumBytes, *StackAlign);
// SPDiff will be negative if this tail call requires more space than we
// would automatically have in our incoming argument space. Positive if we
@@ -4711,8 +4712,9 @@ SDValue ARMTargetLowering::LowerFormalArguments(
if (canGuaranteeTCO(CallConv, TailCallOpt)) {
// The only way to guarantee a tail call is if the callee restores its
// argument area, but it must also keep the stack aligned when doing so.
- const DataLayout &DL = DAG.getDataLayout();
- StackArgSize = alignTo(StackArgSize, DL.getStackAlignment());
+ MaybeAlign StackAlign = DAG.getDataLayout().getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
+ StackArgSize = alignTo(StackArgSize, *StackAlign);
AFI->setArgumentStackToRestore(StackArgSize);
}
@@ -22033,7 +22035,9 @@ Align ARMTargetLowering::getABIAlignmentForCallingConv(
// Avoid over-aligning vector parameters. It would require realigning the
// stack and waste space for no real benefit.
- return std::min(ABITypeAlign, DL.getStackAlignment());
+ MaybeAlign StackAlign = DL.getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
+ return std::min(ABITypeAlign, *StackAlign);
}
/// Return true if a type is an AAPCS-VFP homogeneous aggregate or one of
diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
index 13d3e3e31dd45d..4578ff7f715146 100644
--- a/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
+++ b/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
@@ -1189,8 +1189,9 @@ WebAssemblyTargetLowering::LowerCall(CallLoweringInfo &CLI,
if (IsVarArg && NumBytes) {
// For non-fixed arguments, next emit stores to store the argument values
// to the stack buffer at the offsets computed above.
- int FI = MF.getFrameInfo().CreateStackObject(NumBytes,
- Layout.getStackAlignment(),
+ MaybeAlign StackAlign = Layout.getStackAlignment();
+ assert(StackAlign && "data layout string is missing stack alignment");
+ int FI = MF.getFrameInfo().CreateStackObject(NumBytes, *StackAlign,
/*isSS=*/false);
unsigned ValNo = 0;
SmallVector<SDValue, 8> Chains;
diff --git a/llvm/lib/Transforms/IPO/ExpandVariadics.cpp b/llvm/lib/Transforms/IPO/ExpandVariadics.cpp
index 49bfec297bc173..a7a01ca1055dd3 100644
--- a/llvm/lib/Transforms/IPO/ExpandVariadics.cpp
+++ b/llvm/lib/Transforms/IPO/ExpandVariadics.cpp
@@ -748,10 +748,10 @@ bool ExpandVariadics::expandCall(Module &M, IRBuilder<> &Builder, CallBase *CB,
// This is an awkward way to guess whether there is a known stack alignment
// without hitting an assert in DL.getStackAlignment, 1024 is an arbitrary
// number likely to be greater than the natural stack alignment.
- // TODO: DL.getStackAlignment could return a MaybeAlign instead of assert
Align AllocaAlign = MaxFieldAlign;
- if (DL.exceedsNaturalStackAlignment(Align(1024)))
- AllocaAlign = std::max(AllocaAlign, DL.getStackAlignment());
+ if (MaybeAlign StackAlign = DL.getStackAlignment();
+ StackAlign && *StackAlign > AllocaAlign)
+ AllocaAlign = *StackAlign;
// Put the alloca to hold the variadic args in the entry basic block.
Builder.SetInsertPointPastAllocas(CBF);
diff --git a/llvm/lib/Transforms/Utils/Local.cpp b/llvm/lib/Transforms/Utils/Local.cpp
index d3710de1964ece..e4809cd4bb44db 100644
--- a/llvm/lib/Transforms/Utils/Local.cpp
+++ b/llvm/lib/Transforms/Utils/Local.cpp
@@ -1506,7 +1506,8 @@ Align llvm::tryEnforceAlignment(Value *V, Align PrefAlign,
// If the preferred alignment is greater than the natural stack alignment
// then don't round up. This avoids dynamic stack realignment.
- if (DL.exceedsNaturalStackAlignment(PrefAlign))
+ MaybeAlign StackAlign = DL.getStackAlignment();
+ if (StackAlign && PrefAlign > *StackAlign)
return CurrentAlign;
AI->setAlignment(PrefAlign);
return PrefAlign;
diff --git a/llvm/unittests/IR/DataLayoutTest.cpp b/llvm/unittests/IR/DataLayoutTest.cpp
index 396d44af19f53f..16a603ff6416f4 100644
--- a/llvm/unittests/IR/DataLayoutTest.cpp
+++ b/llvm/unittests/IR/DataLayoutTest.cpp
@@ -444,6 +444,21 @@ TEST(DataLayout, ParseNonIntegralAddrSpace) {
FailedWithMessage("address space 0 cannot be non-integral"));
}
+TEST(DataLayout, GetStackAlignment) {
+ DataLayout Default;
+ EXPECT_FALSE(Default.getStackAlignment().has_value());
+
+ std::pair<StringRef, Align> Cases[] = {
+ {"S8", Align(1)},
+ {"S64", Align(8)},
+ {"S32768", Align(4096)},
+ };
+ for (auto [Layout, Val] : Cases) {
+ DataLayout DL = cantFail(DataLayout::parse(Layout));
+ EXPECT_EQ(DL.getStackAlignment(), Val) << Layout;
+ }
+}
+
TEST(DataLayout, GetPointerSizeInBits) {
std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = {
{"", 64, 64, 64},
More information about the llvm-commits
mailing list