[llvm] f658ca1 - [mlgo] Fix build breaks introduced by includes cleanups
Mircea Trofin via llvm-commits
llvm-commits at lists.llvm.org
Mon Mar 21 13:49:48 PDT 2022
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"));
More information about the llvm-commits
mailing list