Skip to content

Commit 87a3d26

Browse files
jurahuljurahul
committed
[NFCI][LLVM/MLIR] Adopt TrailingObjects convienence API
Adopt `TrailingObjects` convienence API that was added in #138970 in LLVM and MLIR code.
1 parent 802d8d9 commit 87a3d26

File tree

13 files changed

+73
-99
lines changed

13 files changed

+73
-99
lines changed

‎llvm/include/llvm/DebugInfo/BTF/BTF.h‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -304,14 +304,12 @@ enum PatchableRelocKind : uint32_t {
304304
// For CommonType sub-types that are followed by a single entry of
305305
// some type in the binary format.
306306
#define BTF_DEFINE_TAIL(Type, Accessor) \
307-
const Type &Accessor() const { return *getTrailingObjects<Type>(); }
307+
const Type &Accessor() const { return *getTrailingObjects(); }
308308

309309
// For CommonType sub-types that are followed by CommonType::getVlen()
310310
// number of entries of some type in the binary format.
311311
#define BTF_DEFINE_TAIL_ARR(Type, Accessor) \
312-
ArrayRef<Type> Accessor() const { \
313-
return ArrayRef<Type>(getTrailingObjects<Type>(), getVlen()); \
314-
}
312+
ArrayRef<Type> Accessor() const { return getTrailingObjects(getVlen()); }
315313

316314
struct ArrayType final : CommonType,
317315
private TrailingObjects<ArrayType, BTFArray> {

‎llvm/include/llvm/IR/DataLayout.h‎

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,9 @@ inline LLVMTargetDataRef wrap(const DataLayout *P) {
564564

565565
/// Used to lazily calculate structure layout information for a target machine,
566566
/// based on the DataLayout structure.
567-
class StructLayout final : public TrailingObjects<StructLayout, TypeSize> {
567+
class StructLayout final : private TrailingObjects<StructLayout, TypeSize> {
568+
friend TrailingObjects;
569+
568570
TypeSize StructSize;
569571
Align StructAlignment;
570572
unsigned IsPadded : 1;
@@ -586,11 +588,11 @@ class StructLayout final : public TrailingObjects<StructLayout, TypeSize> {
586588
unsigned getElementContainingOffset(uint64_t FixedOffset) const;
587589

588590
MutableArrayRef<TypeSize> getMemberOffsets() {
589-
return llvm::MutableArrayRef(getTrailingObjects<TypeSize>(), NumElements);
591+
return getTrailingObjects(NumElements);
590592
}
591593

592594
ArrayRef<TypeSize> getMemberOffsets() const {
593-
return llvm::ArrayRef(getTrailingObjects<TypeSize>(), NumElements);
595+
return getTrailingObjects(NumElements);
594596
}
595597

596598
TypeSize getElementOffset(unsigned Idx) const {
@@ -606,10 +608,6 @@ class StructLayout final : public TrailingObjects<StructLayout, TypeSize> {
606608
friend class DataLayout; // Only DataLayout can create this class
607609

608610
StructLayout(StructType *ST, const DataLayout &DL);
609-
610-
size_t numTrailingObjects(OverloadToken<TypeSize>) const {
611-
return NumElements;
612-
}
613611
};
614612

615613
// The implementation of this method is provided inline as it is particularly

‎llvm/include/llvm/TableGen/Record.h‎

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ class RecordRecTy final : public RecTy,
258258
void Profile(FoldingSetNodeID &ID) const;
259259

260260
ArrayRef<const Record *> getClasses() const {
261-
return ArrayRef(getTrailingObjects<const Record *>(), NumClasses);
261+
return getTrailingObjects(NumClasses);
262262
}
263263

264264
using const_record_iterator = const Record *const *;
@@ -632,9 +632,7 @@ class BitsInit final : public TypedInit,
632632

633633
const Init *resolveReferences(Resolver &R) const override;
634634

635-
ArrayRef<const Init *> getBits() const {
636-
return ArrayRef(getTrailingObjects<const Init *>(), NumBits);
637-
}
635+
ArrayRef<const Init *> getBits() const { return getTrailingObjects(NumBits); }
638636

639637
const Init *getBit(unsigned Bit) const override { return getBits()[Bit]; }
640638
};
@@ -1026,10 +1024,6 @@ class CondOpInit final : public TypedInit,
10261024
CondOpInit(ArrayRef<const Init *> Conds, ArrayRef<const Init *> Values,
10271025
const RecTy *Type);
10281026

1029-
size_t numTrailingObjects(OverloadToken<Init *>) const {
1030-
return 2*NumConds;
1031-
}
1032-
10331027
public:
10341028
CondOpInit(const CondOpInit &) = delete;
10351029
CondOpInit &operator=(const CondOpInit &) = delete;
@@ -1053,11 +1047,11 @@ class CondOpInit final : public TypedInit,
10531047
const Init *getVal(unsigned Num) const { return getVals()[Num]; }
10541048

10551049
ArrayRef<const Init *> getConds() const {
1056-
return ArrayRef(getTrailingObjects<const Init *>(), NumConds);
1050+
return getTrailingObjects(NumConds);
10571051
}
10581052

10591053
ArrayRef<const Init *> getVals() const {
1060-
return ArrayRef(getTrailingObjects<const Init *>() + NumConds, NumConds);
1054+
return ArrayRef(getTrailingObjects() + NumConds, NumConds);
10611055
}
10621056

10631057
const Init *Fold(const Record *CurRec) const;
@@ -1375,7 +1369,7 @@ class VarDefInit final
13751369
bool args_empty() const { return NumArgs == 0; }
13761370

13771371
ArrayRef<const ArgumentInit *> args() const {
1378-
return ArrayRef(getTrailingObjects<const ArgumentInit *>(), NumArgs);
1372+
return getTrailingObjects(NumArgs);
13791373
}
13801374

13811375
const Init *getBit(unsigned Bit) const override {
@@ -1488,11 +1482,11 @@ class DagInit final
14881482
}
14891483

14901484
ArrayRef<const Init *> getArgs() const {
1491-
return ArrayRef(getTrailingObjects<const Init *>(), NumArgs);
1485+
return getTrailingObjects<const Init *>(NumArgs);
14921486
}
14931487

14941488
ArrayRef<const StringInit *> getArgNames() const {
1495-
return ArrayRef(getTrailingObjects<const StringInit *>(), NumArgs);
1489+
return getTrailingObjects<const StringInit *>(NumArgs);
14961490
}
14971491

14981492
const Init *resolveReferences(Resolver &R) const override;

‎llvm/lib/Bitcode/Reader/BitcodeReader.cpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -559,7 +559,7 @@ class BitcodeConstant final : public Value,
559559
static bool classof(const Value *V) { return V->getValueID() == SubclassID; }
560560

561561
ArrayRef<unsigned> getOperandIDs() const {
562-
return ArrayRef(getTrailingObjects<unsigned>(), NumOperands);
562+
return ArrayRef(getTrailingObjects(), NumOperands);
563563
}
564564

565565
std::optional<ConstantRange> getInRange() const {

‎llvm/lib/IR/AttributeImpl.h‎

Lines changed: 9 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -195,15 +195,12 @@ class StringAttributeImpl final
195195

196196
unsigned KindSize;
197197
unsigned ValSize;
198-
size_t numTrailingObjects(OverloadToken<char>) const {
199-
return KindSize + 1 + ValSize + 1;
200-
}
201198

202199
public:
203200
StringAttributeImpl(StringRef Kind, StringRef Val = StringRef())
204201
: AttributeImpl(StringAttrEntry), KindSize(Kind.size()),
205202
ValSize(Val.size()) {
206-
char *TrailingString = getTrailingObjects<char>();
203+
char *TrailingString = getTrailingObjects();
207204
// Some users rely on zero-termination.
208205
llvm::copy(Kind, TrailingString);
209206
TrailingString[KindSize] = '\0';
@@ -212,10 +209,10 @@ class StringAttributeImpl final
212209
}
213210

214211
StringRef getStringKind() const {
215-
return StringRef(getTrailingObjects<char>(), KindSize);
212+
return StringRef(getTrailingObjects(), KindSize);
216213
}
217214
StringRef getStringValue() const {
218-
return StringRef(getTrailingObjects<char>() + KindSize + 1, ValSize);
215+
return StringRef(getTrailingObjects() + KindSize + 1, ValSize);
219216
}
220217

221218
static size_t totalSizeToAlloc(StringRef Kind, StringRef Val) {
@@ -250,25 +247,22 @@ class ConstantRangeListAttributeImpl final
250247
friend TrailingObjects;
251248

252249
unsigned Size;
253-
size_t numTrailingObjects(OverloadToken<ConstantRange>) const { return Size; }
254250

255251
public:
256252
ConstantRangeListAttributeImpl(Attribute::AttrKind Kind,
257253
ArrayRef<ConstantRange> Val)
258254
: EnumAttributeImpl(ConstantRangeListAttrEntry, Kind), Size(Val.size()) {
259255
assert(Size > 0);
260-
ConstantRange *TrailingCR = getTrailingObjects<ConstantRange>();
261-
llvm::uninitialized_copy(Val, TrailingCR);
256+
llvm::uninitialized_copy(Val, getTrailingObjects());
262257
}
263258

264259
~ConstantRangeListAttributeImpl() {
265-
ConstantRange *TrailingCR = getTrailingObjects<ConstantRange>();
266-
for (unsigned I = 0; I != Size; ++I)
267-
TrailingCR[I].~ConstantRange();
260+
for (ConstantRange &CR : getTrailingObjects(Size))
261+
CR.~ConstantRange();
268262
}
269263

270264
ArrayRef<ConstantRange> getConstantRangeListValue() const {
271-
return ArrayRef(getTrailingObjects<ConstantRange>(), Size);
265+
return getTrailingObjects(Size);
272266
}
273267

274268
static size_t totalSizeToAlloc(ArrayRef<ConstantRange> Val) {
@@ -353,7 +347,7 @@ class AttributeSetNode final
353347

354348
using iterator = const Attribute *;
355349

356-
iterator begin() const { return getTrailingObjects<Attribute>(); }
350+
iterator begin() const { return getTrailingObjects(); }
357351
iterator end() const { return begin() + NumAttrs; }
358352

359353
void Profile(FoldingSetNodeID &ID) const {
@@ -383,9 +377,6 @@ class AttributeListImpl final
383377
/// Union of enum attributes available at any index.
384378
AttributeBitSet AvailableSomewhereAttrs;
385379

386-
// Helper fn for TrailingObjects class.
387-
size_t numTrailingObjects(OverloadToken<AttributeSet>) { return NumAttrSets; }
388-
389380
public:
390381
AttributeListImpl(ArrayRef<AttributeSet> Sets);
391382

@@ -407,7 +398,7 @@ class AttributeListImpl final
407398

408399
using iterator = const AttributeSet *;
409400

410-
iterator begin() const { return getTrailingObjects<AttributeSet>(); }
401+
iterator begin() const { return getTrailingObjects(); }
411402
iterator end() const { return begin() + NumAttrSets; }
412403

413404
void Profile(FoldingSetNodeID &ID) const;

‎llvm/lib/IR/Attributes.cpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1237,7 +1237,7 @@ LLVM_DUMP_METHOD void AttributeSet::dump() const {
12371237
AttributeSetNode::AttributeSetNode(ArrayRef<Attribute> Attrs)
12381238
: NumAttrs(Attrs.size()) {
12391239
// There's memory after the node where we can store the entries in.
1240-
llvm::copy(Attrs, getTrailingObjects<Attribute>());
1240+
llvm::copy(Attrs, getTrailingObjects());
12411241

12421242
for (const auto &I : *this) {
12431243
if (I.isStringAttribute())
@@ -1423,7 +1423,7 @@ AttributeListImpl::AttributeListImpl(ArrayRef<AttributeSet> Sets)
14231423
assert(!Sets.empty() && "pointless AttributeListImpl");
14241424

14251425
// There's memory after the node where we can store the entries in.
1426-
llvm::copy(Sets, getTrailingObjects<AttributeSet>());
1426+
llvm::copy(Sets, getTrailingObjects());
14271427

14281428
// Initialize AvailableFunctionAttrs and AvailableSomewhereAttrs
14291429
// summary bitsets.

‎llvm/lib/Support/TrieRawHashMap.cpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ class TrieSubtrie final
6262
public:
6363
using Slot = LazyAtomicPointer<TrieNode>;
6464

65-
Slot &get(size_t I) { return getTrailingObjects<Slot>()[I]; }
65+
Slot &get(size_t I) { return getTrailingObjects()[I]; }
6666
TrieNode *load(size_t I) { return get(I).load(); }
6767

6868
unsigned size() const { return Size; }
@@ -190,7 +190,7 @@ class ThreadSafeTrieRawHashMapBase::ImplType final
190190
}
191191

192192
// Get the root which is the trailing object.
193-
TrieSubtrie *getRoot() { return getTrailingObjects<TrieSubtrie>(); }
193+
TrieSubtrie *getRoot() { return getTrailingObjects(); }
194194

195195
static void *operator new(size_t Size) { return ::operator new(Size); }
196196
void operator delete(void *Ptr) { ::operator delete(Ptr); }

‎llvm/lib/Transforms/IPO/LowerTypeTests.cpp‎

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -285,8 +285,6 @@ class GlobalTypeMember final : TrailingObjects<GlobalTypeMember, MDNode *> {
285285
// module and its jumptable entry needs to be exported to thinlto backends.
286286
bool IsExported;
287287

288-
size_t numTrailingObjects(OverloadToken<MDNode *>) const { return NTypes; }
289-
290288
public:
291289
static GlobalTypeMember *create(BumpPtrAllocator &Alloc, GlobalObject *GO,
292290
bool IsJumpTableCanonical, bool IsExported,
@@ -297,7 +295,7 @@ class GlobalTypeMember final : TrailingObjects<GlobalTypeMember, MDNode *> {
297295
GTM->NTypes = Types.size();
298296
GTM->IsJumpTableCanonical = IsJumpTableCanonical;
299297
GTM->IsExported = IsExported;
300-
llvm::copy(Types, GTM->getTrailingObjects<MDNode *>());
298+
llvm::copy(Types, GTM->getTrailingObjects());
301299
return GTM;
302300
}
303301

@@ -313,9 +311,7 @@ class GlobalTypeMember final : TrailingObjects<GlobalTypeMember, MDNode *> {
313311
return IsExported;
314312
}
315313

316-
ArrayRef<MDNode *> types() const {
317-
return ArrayRef(getTrailingObjects<MDNode *>(), NTypes);
318-
}
314+
ArrayRef<MDNode *> types() const { return getTrailingObjects(NTypes); }
319315
};
320316

321317
struct ICallBranchFunnel final
@@ -329,13 +325,13 @@ struct ICallBranchFunnel final
329325
Call->CI = CI;
330326
Call->UniqueId = UniqueId;
331327
Call->NTargets = Targets.size();
332-
llvm::copy(Targets, Call->getTrailingObjects<GlobalTypeMember *>());
328+
llvm::copy(Targets, Call->getTrailingObjects());
333329
return Call;
334330
}
335331

336332
CallInst *CI;
337333
ArrayRef<GlobalTypeMember *> targets() const {
338-
return ArrayRef(getTrailingObjects<GlobalTypeMember *>(), NTargets);
334+
return getTrailingObjects(NTargets);
339335
}
340336

341337
unsigned UniqueId;

‎mlir/include/mlir/IR/Operation.h‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -679,8 +679,7 @@ class alignas(8) Operation final
679679
if (numRegions == 0)
680680
return MutableArrayRef<Region>();
681681

682-
auto *regions = getTrailingObjects<Region>();
683-
return {regions, numRegions};
682+
return getTrailingObjects<Region>(numRegions);
684683
}
685684

686685
/// Returns the region held by this operation at position 'index'.
@@ -694,7 +693,7 @@ class alignas(8) Operation final
694693
//===--------------------------------------------------------------------===//
695694

696695
MutableArrayRef<BlockOperand> getBlockOperands() {
697-
return {getTrailingObjects<BlockOperand>(), numSuccs};
696+
return getTrailingObjects<BlockOperand>(numSuccs);
698697
}
699698

700699
// Successor iteration.

0 commit comments

Comments
 (0)