From f658ca1abaccd6e0641ef192383e248d35c3e420 Mon Sep 17 00:00:00 2001 From: Mircea Trofin Date: Mon, 21 Mar 2022 13:48:04 -0700 Subject: [PATCH] [mlgo] Fix build breaks introduced by includes cleanups These were not detected by the build bots because those went quietly offline, too, due to a misconfiguration (fixed since) --- llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp index 6118efd..e01838e 100644 --- a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp +++ b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp @@ -15,6 +15,10 @@ #include "RegAllocGreedy.h" #include "llvm/Analysis/AliasAnalysis.h" #include "llvm/Analysis/MLModelRunner.h" +#if defined(LLVM_HAVE_TF_AOT_REGALLOCEVICTMODEL) || defined(LLVM_HAVE_TF_API) +#include "llvm/Analysis/ModelUnderTrainingRunner.h" +#include "llvm/Analysis/NoInferenceModelRunner.h" +#endif #include "llvm/Analysis/ReleaseModeModelRunner.h" #include "llvm/CodeGen/CalcSpillWeights.h" #include "llvm/CodeGen/LiveRegMatrix.h" @@ -48,6 +52,9 @@ using CompiledModelType = NoopSavedModelImpl; // Options that only make sense in development mode #ifdef LLVM_HAVE_TF_API +#include "RegAllocScore.h" +#include "llvm/Analysis/Utils/TFUtils.h" + static cl::opt TrainingLog( "regalloc-training-log", cl::Hidden, cl::desc("Training log for the register allocator eviction model")); -- 2.7.4