[mlgo] Fix bad merge of #156120 with 0082cf4 (#156134)

mtrofin · web-flow · commit c168ce20f51c · 2025-08-29T19:35:48.000-07:00
diff --git a/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp b/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp
@@ -281,7 +281,7 @@ TrainingLogger::TrainingLogger(StringRef LogFileName,
                                const std::vector<TensorSpec> &FeatureMap)
     : LogFileName(LogFileName), MUTR(MUTR), FeatureMap(FeatureMap) {
   // The first output is the inlining decision.
-  std::vector<TensorSpec> FT(getFeatureMap().begin(), getFeatureMap().end());
+  std::vector<TensorSpec> FT(FeatureMap.begin(), FeatureMap.end());
 
   if (MUTR)
     append_range(FT, MUTR->extraOutputsForLoggingSpecs());
@@ -307,8 +307,7 @@ void TrainingLogger::logInlineEvent(const InlineEvent &Event,
                                     const MLModelRunner &ModelRunner) {
   L->startObservation();
   size_t CurrentFeature = 0;
-  size_t FeatureMapSize = getFeatureMap().size();
-  for (; CurrentFeature < FeatureMapSize; ++CurrentFeature)
+  for (; CurrentFeature < FeatureMap.size(); ++CurrentFeature)
     L->logTensorValue(CurrentFeature,
                       reinterpret_cast<const char *>(
                           ModelRunner.getTensorUntyped(CurrentFeature)));
