diff --git a/llvm/include/llvm/Analysis/InlineAdvisor.h b/llvm/include/llvm/Analysis/InlineAdvisor.h --- a/llvm/include/llvm/Analysis/InlineAdvisor.h +++ b/llvm/include/llvm/Analysis/InlineAdvisor.h @@ -14,7 +14,6 @@ #include #include "llvm/Analysis/InlineCost.h" -#include "llvm/Config/config.h" #include "llvm/IR/PassManager.h" namespace llvm { @@ -204,16 +203,12 @@ Result run(Module &M, ModuleAnalysisManager &MAM) { return Result(M, MAM); } }; -#ifdef LLVM_HAVE_TF_AOT std::unique_ptr getReleaseModeAdvisor(Module &M, ModuleAnalysisManager &MAM); -#endif -#ifdef LLVM_HAVE_TF_API std::unique_ptr getDevelopmentModeAdvisor(Module &M, ModuleAnalysisManager &MAM, std::function GetDefaultAdvice); -#endif // Default (manual policy) decision making helper APIs. Shared with the legacy // pass manager inliner. diff --git a/llvm/include/llvm/Analysis/Utils/TFUtils.h b/llvm/include/llvm/Analysis/Utils/TFUtils.h --- a/llvm/include/llvm/Analysis/Utils/TFUtils.h +++ b/llvm/include/llvm/Analysis/Utils/TFUtils.h @@ -9,7 +9,7 @@ #ifndef LLVM_ANALYSIS_UTILS_TFUTILS_H #define LLVM_ANALYSIS_UTILS_TFUTILS_H -#include "llvm/Config/config.h" +#include "llvm/Config/llvm-config.h" #ifdef LLVM_HAVE_TF_API #include "llvm/IR/LLVMContext.h" diff --git a/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp b/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp --- a/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp +++ b/llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp @@ -11,13 +11,12 @@ // loading of a model from a command line option. // //===----------------------------------------------------------------------===// -#include "llvm/Config/config.h" -#if defined(LLVM_HAVE_TF_API) #include "llvm/Analysis/CallGraph.h" #include "llvm/Analysis/InlineSizeEstimatorAnalysis.h" #include "llvm/Analysis/MLInlineAdvisor.h" #include "llvm/Analysis/Utils/TFUtils.h" +#include "llvm/Config/config.h" #include "llvm/IR/LLVMContext.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/ManagedStatic.h" @@ -26,6 +25,8 @@ using namespace llvm; +#if defined(LLVM_HAVE_TF_API) + static cl::opt TrainingLog( "training-log", cl::Hidden, cl::desc("Path where the development - mode inlining log is saved.")); @@ -486,4 +487,10 @@ return std::make_unique( M, MAM, std::move(Runner), GetDefaultAdvice, IsDoingInference); } +#else +std::unique_ptr llvm::getDevelopmentModeAdvisor( + Module &M, ModuleAnalysisManager &MAM, + std::function GetDefaultAdvice) { + return nullptr; +} #endif // defined(LLVM_HAVE_TF_API) diff --git a/llvm/lib/Analysis/InlineAdvisor.cpp b/llvm/lib/Analysis/InlineAdvisor.cpp --- a/llvm/lib/Analysis/InlineAdvisor.cpp +++ b/llvm/lib/Analysis/InlineAdvisor.cpp @@ -18,6 +18,7 @@ #include "llvm/Analysis/ProfileSummaryInfo.h" #include "llvm/Analysis/TargetLibraryInfo.h" #include "llvm/Analysis/TargetTransformInfo.h" +#include "llvm/Config/config.h" #include "llvm/IR/DebugInfoMetadata.h" #include "llvm/IR/Instructions.h" #include "llvm/Support/CommandLine.h" diff --git a/llvm/lib/Analysis/ReleaseModeModelRunner.cpp b/llvm/lib/Analysis/ReleaseModeModelRunner.cpp --- a/llvm/lib/Analysis/ReleaseModeModelRunner.cpp +++ b/llvm/lib/Analysis/ReleaseModeModelRunner.cpp @@ -10,17 +10,19 @@ // Only inference is supported. // //===----------------------------------------------------------------------===// -#include "llvm/Config/config.h" -#if defined(LLVM_HAVE_TF_AOT) #include "llvm/Analysis/InlineModelFeatureMaps.h" #include "llvm/Analysis/MLInlineAdvisor.h" +#include "llvm/Config/config.h" + +#include +#include + +#if defined(LLVM_HAVE_TF_AOT) // codegen-ed file #include "InlinerSizeModel.h" // NOLINT -#include -#include using namespace llvm; namespace { @@ -87,4 +89,11 @@ auto AOTRunner = std::make_unique(M.getContext()); return std::make_unique(M, MAM, std::move(AOTRunner)); } +#else +using namespace llvm; + +std::unique_ptr +llvm::getReleaseModeAdvisor(Module &M, ModuleAnalysisManager &MAM) { + return nullptr; +} #endif // defined(LLVM_HAVE_TF_AOT)