Skip to content

Commit 0082cf4

Browse files
[MLGO] Fix Development Build post f4665b
This broke the development build which is only used by us and build on the bots.
1 parent 3cc1031 commit 0082cf4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ static const std::vector<TensorSpec> TrainingOnlyFeatures{
260260

261261
static const std::vector<TensorSpec> getInputFeatures() {
262262
std::vector<TensorSpec> InputSpecs;
263-
for (const auto &Feature : FeatureMap)
263+
for (const auto &Feature : getFeatureMap())
264264
InputSpecs.push_back(TensorSpec(TFFeedPrefix + Feature.name(), Feature));
265265
append_range(InputSpecs, TrainingOnlyFeatures);
266266
return InputSpecs;
@@ -272,7 +272,7 @@ TrainingLogger::TrainingLogger(StringRef LogFileName,
272272
const ModelUnderTrainingRunner *MUTR)
273273
: LogFileName(LogFileName), MUTR(MUTR) {
274274
// The first output is the inlining decision.
275-
std::vector<TensorSpec> FT(FeatureMap.begin(), FeatureMap.end());
275+
std::vector<TensorSpec> FT(getFeatureMap().begin(), getFeatureMap().end());
276276

277277
if (MUTR)
278278
append_range(FT, MUTR->extraOutputsForLoggingSpecs());
@@ -298,7 +298,7 @@ void TrainingLogger::logInlineEvent(const InlineEvent &Event,
298298
const MLModelRunner &ModelRunner) {
299299
L->startObservation();
300300
size_t CurrentFeature = 0;
301-
size_t FeatureMapSize = FeatureMap.size();
301+
size_t FeatureMapSize = getFeatureMap().size();
302302
for (; CurrentFeature < FeatureMapSize; ++CurrentFeature)
303303
L->logTensorValue(CurrentFeature,
304304
reinterpret_cast<const char *>(

0 commit comments

Comments
 (0)