MLInlineAdvisor.cpp File Reference

#include "llvm/Analysis/MLInlineAdvisor.h"
#include "AnalysisOptions.h"
#include "llvm/ADT/SCCIterator.h"
#include "llvm/Analysis/AssumptionCache.h"
#include "llvm/Analysis/BlockFrequencyInfo.h"
#include "llvm/Analysis/CallGraph.h"
#include "llvm/Analysis/FunctionPropertiesAnalysis.h"
#include "llvm/Analysis/InlineCost.h"
#include "llvm/Analysis/InlineModelFeatureMaps.h"
#include "llvm/Analysis/LazyCallGraph.h"
#include "llvm/Analysis/LoopInfo.h"
#include "llvm/Analysis/MLModelRunner.h"
#include "llvm/Analysis/OptimizationRemarkEmitter.h"
#include "llvm/Analysis/ProfileSummaryInfo.h"
#include "llvm/Analysis/ReleaseModeModelRunner.h"
#include "llvm/Analysis/TargetTransformInfo.h"
#include "llvm/Analysis/TensorSpec.h"
#include "llvm/Analysis/Utils/MLGOUtils.h"
#include "llvm/IR/Dominators.h"
#include "llvm/IR/InstIterator.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/PassManager.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Analysis/EmitCModelRunner.h"
#include "llvm/Analysis/InlinerModels.h"
#include "llvm/Analysis/InlinerModels.def"

Go to the source code of this file.

Macros
#define	MLGO_MODEL(CLASS_NAME, CLI_FLAG)
#define	MLGO_MODEL(CLASS_NAME, CLI_FLAG)
#define	MLGO_MODEL(CLASS_NAME, CLI_FLAG)
#define	DEBUG_TYPE   "inline-ml"
#define	POPULATE_NAMES(DTYPE, SHAPE, NAME, __)

Typedefs
using	CompiledModelType = NoopSavedModelImpl

Enumerations
enum class	EmitCModelChoice { Default }

Functions
static std::unique_ptr< MLModelRunner >	createEmitCModelRunner (LLVMContext &Ctx, const std::vector< TensorSpec > &InputFeatures)
CallBase *	getInlinableCS (Instruction &I)

Variables
static llvm::cl::opt< EmitCModelChoice >	SelectedMLGOModel ("mlgo-model", llvm::cl::desc("Select the MLGO model to execute:"), llvm::cl::init(EmitCModelChoice::Default), llvm::cl::values(clEnumValN(EmitCModelChoice::Default, "default", "Use standard heuristic") #define MLGO_MODEL(CLASS_NAME, CLI_FLAG) \ \‍))

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "inline-ml"

Definition at line 102 of file MLInlineAdvisor.cpp.

MLGO_MODEL [1/3]

#define MLGO_MODEL	(		CLASS_NAME,
			CLI_FLAG )

Value:

CLASS_NAME,

Definition at line 53 of file MLInlineAdvisor.cpp.

MLGO_MODEL [2/3]

#define MLGO_MODEL	(		CLASS_NAME,
			CLI_FLAG )

Value:

  , clEnumValN(EmitCModelChoice::CLASS_NAME, CLI_FLAG,                         \
               "Use the " CLI_FLAG " MLGO model")

Definition at line 53 of file MLInlineAdvisor.cpp.

MLGO_MODEL [3/3]

#define MLGO_MODEL	(		CLASS_NAME,
			CLI_FLAG )

Value:

  case EmitCModelChoice::CLASS_NAME:                                           \
    return std::make_unique<EmitCModelRunner<CLASS_NAME>>(Ctx, InputFeatures);

Definition at line 53 of file MLInlineAdvisor.cpp.

POPULATE_NAMES

#define POPULATE_NAMES	(		DTYPE,
			SHAPE,
			NAME,
			__ )

Value:

TensorSpec::createSpec<DTYPE>(#NAME, SHAPE),

Referenced by llvm::MLInlineAdvisor::getInitialFeatureMap().

Typedef Documentation

CompiledModelType

using CompiledModelType = NoopSavedModelImpl

Definition at line 45 of file MLInlineAdvisor.cpp.

Enumeration Type Documentation

EmitCModelChoice

enum class EmitCModelChoice

strong

Enumerator
Default

Definition at line 51 of file MLInlineAdvisor.cpp.

Function Documentation

createEmitCModelRunner()

std::unique_ptr< MLModelRunner > createEmitCModelRunner ( LLVMContext & Ctx, const std::vector< TensorSpec > & InputFeatures )

static

Definition at line 69 of file MLInlineAdvisor.cpp.

References Default, llvm::InputFeatures, llvm_unreachable, and SelectedMLGOModel.

Referenced by llvm::getReleaseModeAdvisor().

getInlinableCS()

CallBase * getInlinableCS

(

Instruction &

I

)

Definition at line 127 of file MLInlineAdvisor.cpp.

References llvm::dyn_cast(), and I.

Referenced by llvm::MLInlineAdvisor::MLInlineAdvisor().

Variable Documentation

SelectedMLGOModel

llvm::cl::opt< EmitCModelChoice > SelectedMLGOModel("mlgo-model", llvm::cl::desc("Select the MLGO model to execute:"), llvm::cl::init(EmitCModelChoice::Default), llvm::cl::values(clEnumValN(EmitCModelChoice::Default, "default", "Use standard heuristic") #define MLGO_MODEL(CLASS_NAME, CLI_FLAG) \ \‍)) ( "mlgo-model" , llvm::cl::desc("Select the MLGO model to execute:") , llvm::cl::init(EmitCModelChoice::Default) , llvm::cl::values(clEnumValN(EmitCModelChoice::Default, "default", "Use standard heuristic")#define MLGO_MODEL(CLASS_NAME, CLI_FLAG) \ \‍)  )

static

Referenced by createEmitCModelRunner(), and llvm::getReleaseModeAdvisor().