diff --git a/llvm/lib/Analysis/InlineCost.cpp b/llvm/lib/Analysis/InlineCost.cpp --- a/llvm/lib/Analysis/InlineCost.cpp +++ b/llvm/lib/Analysis/InlineCost.cpp @@ -126,6 +126,10 @@ InstrCost("inline-instr-cost", cl::Hidden, cl::init(5), cl::desc("Cost of a single instruction when inlining")); +static cl::opt + MemAccessCost("inline-memaccess-cost", cl::Hidden, cl::init(0), + cl::desc("Cost of load/store instruction when inlining")); + static cl::opt CallPenalty( "inline-call-penalty", cl::Hidden, cl::init(25), cl::desc("Call penalty that is applied per callsite when inlining")); @@ -282,6 +286,9 @@ /// Called to account for a call. virtual void onCallPenalty() {} + /// Called to account for a load or store. + virtual void onMemAccess(){}; + /// Called to account for the expectation the inlining would result in a load /// elimination. virtual void onLoadEliminationOpportunity() {} @@ -625,6 +632,9 @@ } void onCallPenalty() override { addCost(CallPenalty); } + + void onMemAccess() override { addCost(MemAccessCost); } + void onCallArgumentSetup(const CallBase &Call) override { // Pay the price of the argument setup. We account for the average 1 // instruction per call argument setup here. @@ -2044,6 +2054,7 @@ return true; } + onMemAccess(); return false; } @@ -2060,6 +2071,8 @@ // 2. We should probably at some point thread MemorySSA for the callee into // this and then use that to actually compute *really* precise savings. disableLoadElimination(); + + onMemAccess(); return false; }