[llvm] 4a4444c - [CodeModel] Factor getCost out of CostModelPrinter loop. NFC

[David Green via llvm-commits]

Author: David Green
Date: 2025-03-11T07:58:29Z
New Revision: 4a4444c0b2f68bec1db8e2cc8d133982d5a339e3

URL: https://github.com/llvm/llvm-project/commit/4a4444c0b2f68bec1db8e2cc8d133982d5a339e3
DIFF: https://github.com/llvm/llvm-project/commit/4a4444c0b2f68bec1db8e2cc8d133982d5a339e3.diff

LOG: [CodeModel] Factor getCost out of CostModelPrinter loop. NFC

This helps in a follow-up so that it can be called multiple times with
different cost types.

Added: 
    

Modified: 
    llvm/lib/Analysis/CostModel.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Analysis/CostModel.cpp b/llvm/lib/Analysis/CostModel.cpp
index 27a946c846e67..b863bf77c6ddd 100644
--- a/llvm/lib/Analysis/CostModel.cpp
+++ b/llvm/lib/Analysis/CostModel.cpp
@@ -63,6 +63,22 @@ static cl::opt<IntrinsicCostStrategy> IntrinsicCost(
 #define CM_NAME "cost-model"
 #define DEBUG_TYPE CM_NAME
 
+static InstructionCost getCost(Instruction &Inst, TTI::TargetCostKind CostKind,
+                               TargetTransformInfo &TTI,
+                               TargetLibraryInfo &TLI) {
+  auto *II = dyn_cast<IntrinsicInst>(&Inst);
+  if (II && IntrinsicCost != IntrinsicCostStrategy::InstructionCost) {
+    IntrinsicCostAttributes ICA(
+        II->getIntrinsicID(), *II, InstructionCost::getInvalid(),
+        /*TypeBasedOnly=*/IntrinsicCost ==
+            IntrinsicCostStrategy::TypeBasedIntrinsicCost,
+        &TLI);
+    return TTI.getIntrinsicInstrCost(ICA, CostKind);
+  }
+
+  return TTI.getInstructionCost(&Inst, CostKind);
+}
+
 PreservedAnalyses CostModelPrinterPass::run(Function &F,
                                             FunctionAnalysisManager &AM) {
   auto &TTI = AM.getResult<TargetIRAnalysis>(F);
@@ -72,19 +88,7 @@ PreservedAnalyses CostModelPrinterPass::run(Function &F,
     for (Instruction &Inst : B) {
       // TODO: Use a pass parameter instead of cl::opt CostKind to determine
       // which cost kind to print.
-      InstructionCost Cost;
-      auto *II = dyn_cast<IntrinsicInst>(&Inst);
-      if (II && IntrinsicCost != IntrinsicCostStrategy::InstructionCost) {
-        IntrinsicCostAttributes ICA(
-            II->getIntrinsicID(), *II, InstructionCost::getInvalid(),
-            /*TypeBasedOnly=*/IntrinsicCost ==
-                IntrinsicCostStrategy::TypeBasedIntrinsicCost,
-            &TLI);
-        Cost = TTI.getIntrinsicInstrCost(ICA, CostKind);
-      } else {
-        Cost = TTI.getInstructionCost(&Inst, CostKind);
-      }
-
+      InstructionCost Cost = getCost(Inst, CostKind, TTI, TLI);
       if (auto CostVal = Cost.getValue())
         OS << "Cost Model: Found an estimated cost of " << *CostVal;
       else