MLInlineAdvisor.cpp File Reference

#include "llvm/Analysis/MLInlineAdvisor.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"

Go to the source code of this file.

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

Typedefs
using	CompiledModelType = NoopSavedModelImpl

Enumerations
enum class	SkipMLPolicyCriteria { Never , IfCallerIsNotCold }
enum class	EmitCModelChoice { Default }

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

Variables
static cl::opt< std::string >	InteractiveChannelBaseName ("inliner-interactive-channel-base", cl::Hidden, cl::desc("Base file path for the interactive mode. The incoming filename should " "have the name <inliner-interactive-channel-base>.in, while the " "outgoing name should be <inliner-interactive-channel-base>.out"))
static const std::string	InclDefaultMsg
static cl::opt< bool >	InteractiveIncludeDefault ("inliner-interactive-include-default", cl::Hidden, cl::desc(InclDefaultMsg))
static cl::opt< SkipMLPolicyCriteria >	SkipPolicy ("ml-inliner-skip-policy", cl::Hidden, cl::init(SkipMLPolicyCriteria::Never), cl::values(clEnumValN(SkipMLPolicyCriteria::Never, "never", "never"), clEnumValN(SkipMLPolicyCriteria::IfCallerIsNotCold, "if-caller-not-cold", "if the caller is not cold")))
static cl::opt< std::string >	ModelSelector ("ml-inliner-model-selector", cl::Hidden, cl::init(""))
static cl::opt< bool >	StopImmediatelyForTest ("ml-inliner-stop-immediately", cl::Hidden)
constexpr bool	HaveMLIRLoweringInliner = false
static const EmitCModelChoice	SelectedMLGOModel = EmitCModelChoice::Default
static cl::opt< float >	SizeIncreaseThreshold ("ml-advisor-size-increase-threshold", cl::Hidden, cl::desc("Maximum factor by which expected native size may increase before " "blocking any further inlining."), cl::init(2.0))
static cl::opt< bool >	KeepFPICache ("ml-advisor-keep-fpi-cache", cl::Hidden, cl::desc("For test - keep the ML Inline advisor's FunctionPropertiesInfo cache"), cl::init(false))

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "inline-ml"

Definition at line 138 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 72 of file MLInlineAdvisor.cpp.

Enumeration Type Documentation

EmitCModelChoice

enum class EmitCModelChoice

strong

Enumerator
Default

Definition at line 112 of file MLInlineAdvisor.cpp.

SkipMLPolicyCriteria

enum class SkipMLPolicyCriteria

strong

Enumerator
Never
IfCallerIsNotCold

Definition at line 53 of file MLInlineAdvisor.cpp.

Function Documentation

createEmitCModelRunner()

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

inlinestatic

Definition at line 115 of file MLInlineAdvisor.cpp.

Referenced by llvm::getReleaseModeAdvisor().

getInlinableCS()

CallBase * getInlinableCS

(

Instruction &

I

)

Definition at line 175 of file MLInlineAdvisor.cpp.

References llvm::dyn_cast(), and I.

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

Variable Documentation

HaveMLIRLoweringInliner

bool HaveMLIRLoweringInliner = false

constexpr

Definition at line 111 of file MLInlineAdvisor.cpp.

Referenced by llvm::getReleaseModeAdvisor().

InclDefaultMsg

const std::string InclDefaultMsg

static

Initial value:

=
    (Twine("In interactive mode, also send the default policy decision: ") +
     DefaultDecisionName + ".")
        .str()

Definition at line 45 of file MLInlineAdvisor.cpp.

InteractiveChannelBaseName

cl::opt< std::string > InteractiveChannelBaseName("inliner-interactive-channel-base", cl::Hidden, cl::desc( "Base file path for the interactive mode. The incoming filename should " "have the name <inliner-interactive-channel-base>.in, while the " "outgoing name should be <inliner-interactive-channel-base>.out")) ( "inliner-interactive-channel-base" , cl::Hidden , cl::desc( "Base file path for the interactive mode. The incoming filename should " "have the name <inliner-interactive-channel-base>.in, while the " "outgoing name should be <inliner-interactive-channel-base>.out")  )

static

Referenced by llvm::createReleaseModeAdvisorAnalysisLegacy(), llvm::createReleaseModeAdvisorProvider(), llvm::createReleaseModeModelRunner(), llvm::createReleaseModePriorityAdvisorAnalysis(), llvm::MLInlineAdvisor::getAdviceImpl(), llvm::ReleaseModePriorityAdvisorProvider::getAdvisor(), llvm::getReleaseModeAdvisor(), INITIALIZE_PASS(), llvm::isReleaseModelValid(), and llvm::isReleaseModelValid().

InteractiveIncludeDefault

cl::opt< bool > InteractiveIncludeDefault("inliner-interactive-include-default", cl::Hidden, cl::desc(InclDefaultMsg)) ( "inliner-interactive-include-default" , cl::Hidden , cl::desc(InclDefaultMsg)  )

static

Referenced by llvm::MLInlineAdvisor::getAdviceImpl(), and llvm::MLInlineAdvisor::MLInlineAdvisor().

KeepFPICache

cl::opt< bool > KeepFPICache("ml-advisor-keep-fpi-cache", cl::Hidden, cl::desc( "For test - keep the ML Inline advisor's FunctionPropertiesInfo cache"), cl::init(false)) ( "ml-advisor-keep-fpi-cache" , cl::Hidden , cl::desc( "For test - keep the ML Inline advisor's FunctionPropertiesInfo cache") , cl::init(false)  )

static

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

ModelSelector

cl::opt< std::string > ModelSelector("ml-inliner-model-selector", cl::Hidden, cl::init("")) ( "ml-inliner-model-selector" , cl::Hidden , cl::init("")  )

static

Referenced by llvm::getReleaseModeAdvisor().

SelectedMLGOModel

const EmitCModelChoice SelectedMLGOModel = EmitCModelChoice::Default

static

Definition at line 113 of file MLInlineAdvisor.cpp.

Referenced by llvm::getReleaseModeAdvisor().

SizeIncreaseThreshold

cl::opt< float > SizeIncreaseThreshold("ml-advisor-size-increase-threshold", cl::Hidden, cl::desc("Maximum factor by which expected native size may increase before " "blocking any further inlining."), cl::init(2.0)) ( "ml-advisor-size-increase-threshold" , cl::Hidden , cl::desc("Maximum factor by which expected native size may increase before " "blocking any further inlining.") , cl::init(2.0)  )

static

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

SkipPolicy

cl::opt< SkipMLPolicyCriteria > SkipPolicy("ml-inliner-skip-policy", cl::Hidden, cl::init(SkipMLPolicyCriteria::Never), cl::values(clEnumValN(SkipMLPolicyCriteria::Never, "never", "never"), clEnumValN(SkipMLPolicyCriteria::IfCallerIsNotCold, "if-caller-not-cold", "if the caller is not cold"))) ( "ml-inliner-skip-policy" , cl::Hidden , cl::init(SkipMLPolicyCriteria::Never) , cl::values(clEnumValN(SkipMLPolicyCriteria::Never, "never", "never"), clEnumValN(SkipMLPolicyCriteria::IfCallerIsNotCold, "if-caller-not-cold", "if the caller is not cold"))  )

static

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

StopImmediatelyForTest

cl::opt< bool > StopImmediatelyForTest("ml-inliner-stop-immediately", cl::Hidden) ( "ml-inliner-stop-immediately" , cl::Hidden  )

static

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