Skip to content

Commit 3ff9c45

Browse files
committed
address reviews, use #else
1 parent 3606cc1 commit 3ff9c45

File tree

3 files changed

+17
-18
lines changed

3 files changed

+17
-18
lines changed

llvm/include/llvm/CodeGen/RegAllocEvictionAdvisor.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include "llvm/IR/PassManager.h"
2121
#include "llvm/MC/MCRegister.h"
2222
#include "llvm/Pass.h"
23+
#include "llvm/Support/Compiler.h"
2324

2425
namespace llvm {
2526
class AllocationOrder;
@@ -279,7 +280,7 @@ RegAllocEvictionAdvisorAnalysisLegacy *createReleaseModeAdvisorAnalysisLegacy();
279280
RegAllocEvictionAdvisorAnalysisLegacy *
280281
createDevelopmentModeAdvisorAnalysisLegacy();
281282

282-
RegAllocEvictionAdvisorProvider *
283+
LLVM_ATTRIBUTE_RETURNS_NONNULL RegAllocEvictionAdvisorProvider *
283284
createReleaseModeAdvisorProvider(LLVMContext &Ctx);
284285

285286
RegAllocEvictionAdvisorProvider *

llvm/lib/CodeGen/MLRegAllocEvictAdvisor.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -620,9 +620,6 @@ class DevelopmentModeEvictionAdvisorAnalysisLegacy final
620620
AU.addRequired<MachineLoopInfoWrapperPass>();
621621
RegAllocEvictionAdvisorAnalysisLegacy::getAnalysisUsage(AU);
622622
}
623-
624-
private:
625-
// std::unique_ptr<DevelopmentModeEvictionAdvisorProvider> Provider;
626623
};
627624

628625
#endif // #ifdef LLVM_HAVE_TFLITE

llvm/lib/CodeGen/RegAllocEvictionAdvisor.cpp

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -122,20 +122,20 @@ void RegAllocEvictionAdvisorAnalysis::initializeProvider(
122122
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Default:
123123
Provider.reset(
124124
new DefaultEvictionAdvisorProvider(/*NotAsRequested=*/false, Ctx));
125-
break;
125+
return;
126126
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Development:
127127
#if defined(LLVM_HAVE_TFLITE)
128128
Provider.reset(createDevelopmentModeAdvisorProvider(Ctx));
129+
#else
130+
Provider.reset(
131+
new DefaultEvictionAdvisorProvider(/*NotAsRequested=*/true, Ctx));
129132
#endif
130-
break;
133+
assert(Provider && "EvictionAdvisorProvider cannot be null");
134+
return;
131135
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Release:
132136
Provider.reset(createReleaseModeAdvisorProvider(Ctx));
133-
break;
137+
return;
134138
}
135-
136-
if (!Provider)
137-
Provider.reset(
138-
new DefaultEvictionAdvisorProvider(/*NotAsRequested=*/true, Ctx));
139139
}
140140

141141
RegAllocEvictionAdvisorAnalysis::Result
@@ -151,20 +151,21 @@ Pass *llvm::callDefaultCtor<RegAllocEvictionAdvisorAnalysisLegacy>() {
151151
Pass *Ret = nullptr;
152152
switch (Mode) {
153153
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Default:
154-
Ret = new DefaultEvictionAdvisorAnalysisLegacy(/*NotAsRequested*/ false);
154+
return new DefaultEvictionAdvisorAnalysisLegacy(/*NotAsRequested*/ false);
155+
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Release:
156+
Ret = createReleaseModeAdvisorAnalysisLegacy();
155157
break;
156158
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Development:
157159
#if defined(LLVM_HAVE_TFLITE)
158160
Ret = createDevelopmentModeAdvisorAnalysisLegacy();
159161
#endif
160162
break;
161-
case RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Release:
162-
Ret = createReleaseModeAdvisorAnalysisLegacy();
163-
break;
164163
}
165-
if (Ret)
166-
return Ret;
167-
return new DefaultEvictionAdvisorAnalysisLegacy(/*NotAsRequested*/ true);
164+
165+
// release or development mode advisor may not be supported
166+
if (!Ret)
167+
Ret = new DefaultEvictionAdvisorAnalysisLegacy(/*NotAsRequested*/ true);
168+
return Ret;
168169
}
169170

170171
StringRef RegAllocEvictionAdvisorAnalysisLegacy::getPassName() const {

0 commit comments

Comments
 (0)