[llvm] f658ca1 - [mlgo] Fix build breaks introduced by includes cleanups

[Mircea Trofin via llvm-commits]

Author: Mircea Trofin
Date: 2022-03-21T13:49:40-07:00
New Revision: f658ca1abaccd6e0641ef192383e248d35c3e420

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

LOG: [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)

Added: 
    

Modified: 
    llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp
index 6118efdbd0b4d..e01838e5fe115 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<std::string> TrainingLog(
     "regalloc-training-log", cl::Hidden,
     cl::desc("Training log for the register allocator eviction model"));