return I->second.get();
}
- void logRewardIfNeeded(const MachineFunction &MF, float Reward) override {
+ void logRewardIfNeeded(const MachineFunction &MF,
+ llvm::function_ref<float()> GetReward) override {
if (auto *Log = this->getLogger(MF))
- Log->logFloatFinalReward(Reward);
+ Log->logFloatFinalReward(GetReward());
}
private:
}
bool RegAllocScoring::runOnMachineFunction(MachineFunction &MF) {
- float Reward = static_cast<float>(
- calculateRegAllocScore(MF, getAnalysis<MachineBlockFrequencyInfo>())
- .getScore());
-
- getAnalysis<RegAllocEvictionAdvisorAnalysis>().logRewardIfNeeded(MF, Reward);
- getAnalysis<RegAllocPriorityAdvisorAnalysis>().logRewardIfNeeded(MF, Reward);
-
+ Optional<float> CachedReward;
+ auto GetReward = [&]() {
+ if (!CachedReward)
+ CachedReward = static_cast<float>(
+ calculateRegAllocScore(MF, getAnalysis<MachineBlockFrequencyInfo>())
+ .getScore());
+ return *CachedReward;
+ };
+
+ getAnalysis<RegAllocEvictionAdvisorAnalysis>().logRewardIfNeeded(MF,
+ GetReward);
+ getAnalysis<RegAllocPriorityAdvisorAnalysis>().logRewardIfNeeded(MF,
+ GetReward);
return false;
}
#endif // #ifdef LLVM_HAVE_TF_API
virtual std::unique_ptr<RegAllocEvictionAdvisor>
getAdvisor(const MachineFunction &MF, const RAGreedy &RA) = 0;
AdvisorMode getAdvisorMode() const { return Mode; }
- virtual void logRewardIfNeeded(const MachineFunction &MF, float Reward){};
+ virtual void logRewardIfNeeded(const MachineFunction &MF,
+ llvm::function_ref<float()> GetReward){};
protected:
// This analysis preserves everything, and subclasses may have additional
virtual std::unique_ptr<RegAllocPriorityAdvisor>
getAdvisor(const MachineFunction &MF, const RAGreedy &RA) = 0;
AdvisorMode getAdvisorMode() const { return Mode; }
- virtual void logRewardIfNeeded(const MachineFunction &MF, float Reward){};
+ virtual void logRewardIfNeeded(const MachineFunction &MF,
+ llvm::function_ref<float()> GetReward){};
protected:
// This analysis preserves everything, and subclasses may have additional