Skip to content

Commit 435ce05

Browse files
committed
Fix formatting, tweak use count.
1 parent 6b368d5 commit 435ce05

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

llvm/lib/Target/AMDGPU/AMDGPUExpandFeaturePredicates.cpp

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -134,8 +134,9 @@ static inline SmallVector<Function *> collectUsedFunctions(Module &M) {
134134
continue;
135135
if (!F.hasInternalLinkage() && !F.hasPrivateLinkage())
136136
continue;
137-
if (F.hasNUndroppableUsesOrMore(1))
138-
Ret.push_back(&F);
137+
if (F.hasNUndroppableUses(0))
138+
continue;
139+
Ret.push_back(&F);
139140
}
140141

141142
return Ret;
@@ -147,8 +148,12 @@ static inline void removeUnreachable(const Container0 &Predicates,
147148
const Container2 &UnreachableFns) {
148149
for_each(Predicates, [](auto &&P) { P->eraseFromParent(); });
149150
for_each(PredicatedFns, [](auto &&F) { removeUnreachableBlocks(*F); });
150-
for_each(UnreachableFns,
151-
[](auto &&F) { if (F->getNumUses() == 0) F->eraseFromParent(); });
151+
for_each(UnreachableFns, [](auto &&F) {
152+
if (!F->hasNUndroppableUses(0))
153+
return;
154+
F->dropDroppableUses();
155+
F->eraseFromParent();
156+
});
152157
}
153158

154159
PreservedAnalyses

0 commit comments

Comments
 (0)