#include "MLIR/Interfaces/GradientUtils.h"

Inheritance diagram for mlir::enzyme::MGradientUtils:

[legend]

Collaboration diagram for mlir::enzyme::MGradientUtils:

[legend]

Public Member Functions
SmallVector< mlir::Value, 1 >	getNewFromOriginal (ValueRange originst) const

mlir::Value	getNewFromOriginal (const mlir::Value originst) const

mlir::Block *	getNewFromOriginal (mlir::Block *originst) const

Operation *	getNewFromOriginal (Operation *originst) const

	MGradientUtils (MEnzymeLogic &Logic, FunctionOpInterface newFunc_, FunctionOpInterface oldFunc_, MTypeAnalysis &TA_, MTypeResults TR_, IRMapping &invertedPointers_, const llvm::ArrayRef< bool > returnPrimals, const llvm::ArrayRef< bool > returnShadows, const SmallPtrSetImpl< mlir::Value > &constantvalues_, const SmallPtrSetImpl< mlir::Value > &activevals_, ArrayRef< DIFFE_TYPE > ReturnActivities, ArrayRef< DIFFE_TYPE > ArgDiffeTypes_, IRMapping &originalToNewFn_, std::map< Operation , Operation > &originalToNewFnOps_, DerivativeMode mode, unsigned width, bool omp, llvm::StringRef postpasses, bool verifyPostPasses, bool strongZero)

void	erase (Operation *op)

void	replaceOrigOpWith (Operation *op, ValueRange vals)

void	eraseIfUnused (Operation *op, bool erase=true, bool check=true)

bool	isConstantInstruction (mlir::Operation *v) const

bool	isConstantValue (mlir::Value v) const

mlir::Value	invertPointerM (mlir::Value v, OpBuilder &Builder2)

void	forceAugmentedReturns ()

Operation *	cloneWithNewOperands (OpBuilder &B, Operation *op)

LogicalResult	visitChild (Operation *op)

void	setDiffe (mlir::Value origv, mlir::Value newv, mlir::OpBuilder &builder)

void	setInvertedPointer (mlir::Value origv, mlir::Value newv)

mlir::Type	getShadowType (mlir::Type T)

Static Public Member Functions
static llvm::SmallVector< mlir::Value, 1 >	reindex_arguments (llvm::ArrayRef< mlir::Value > vals, mlir::OperandRange range)

Public Attributes
FunctionOpInterface	newFunc

MEnzymeLogic &	Logic

bool	AtomicAdd

DerivativeMode	mode

FunctionOpInterface	oldFunc

IRMapping	invertedPointers

IRMapping	originalToNewFn

std::map< Operation , Operation >	originalToNewFnOps

SmallPtrSet< Block *, 4 >	blocksNotForAnalysis

DenseMap< Operation *, bool >	readOnlyCache

std::unique_ptr< enzyme::ActivityAnalyzer >	activityAnalyzer

MTypeAnalysis &	TA

MTypeResults	TR

bool	omp

bool	verifyPostPasses

llvm::StringRef	postpasses

bool	strongZero

const llvm::ArrayRef< bool >	returnPrimals

const llvm::ArrayRef< bool >	returnShadows

unsigned	width

ArrayRef< DIFFE_TYPE >	ArgDiffeTypes

ArrayRef< DIFFE_TYPE >	RetDiffeTypes

Detailed Description

Definition at line 20 of file GradientUtils.h.

Constructor & Destructor Documentation

◆ MGradientUtils()

mlir::enzyme::MGradientUtils::MGradientUtils	(	MEnzymeLogic &	Logic,
		FunctionOpInterface	newFunc_,
		FunctionOpInterface	oldFunc_,
		MTypeAnalysis &	TA_,
		MTypeResults	TR_,
		IRMapping &	invertedPointers_,
		const llvm::ArrayRef< bool >	returnPrimals,
		const llvm::ArrayRef< bool >	returnShadows,
		const SmallPtrSetImpl< mlir::Value > &	constantvalues_,
		const SmallPtrSetImpl< mlir::Value > &	activevals_,
		ArrayRef< DIFFE_TYPE >	ReturnActivities,
		ArrayRef< DIFFE_TYPE >	ArgDiffeTypes_,
		IRMapping &	originalToNewFn_,
		std::map< Operation , Operation > &	originalToNewFnOps_,
		DerivativeMode	mode,
		unsigned	width,
		bool	omp,
		llvm::StringRef	postpasses,
		bool	verifyPostPasses,
		bool	strongZero )

Definition at line 30 of file GradientUtils.cpp.

Member Function Documentation

◆ cloneWithNewOperands()

Operation * mlir::enzyme::MGradientUtils::cloneWithNewOperands	(	OpBuilder &	B,
		Operation *	op )

Definition at line 102 of file GradientUtils.cpp.

◆ erase()

void mlir::enzyme::MGradientUtils::erase ( Operation * op )

inline

Definition at line 69 of file GradientUtils.h.

Referenced by mlir::enzyme::detail::branchingForwardHandler(), mlir::enzyme::detail::controlFlowForwardHandler(), AutoDiffCallFwd::createForwardModeTangent(), createTerminator(), and handleReturns().

◆ eraseIfUnused()

void mlir::enzyme::MGradientUtils::eraseIfUnused	(	Operation *	op,
		bool	erase = true,
		bool	check = true )

inline

Definition at line 78 of file GradientUtils.h.

Referenced by mlir::enzyme::detail::allocationForwardHandler().

◆ forceAugmentedReturns()

void mlir::enzyme::MGradientUtils::forceAugmentedReturns ( )

Definition at line 251 of file GradientUtils.cpp.

Referenced by mlir::enzyme::MEnzymeLogic::CreateReverseDiff().

◆ getNewFromOriginal() [1/4]

mlir::Value mlir::enzyme::MGradientUtils::getNewFromOriginal ( const mlir::Value originst ) const

Definition at line 54 of file GradientUtils.cpp.

◆ getNewFromOriginal() [2/4]

Block * mlir::enzyme::MGradientUtils::getNewFromOriginal ( mlir::Block * originst ) const

Definition at line 75 of file GradientUtils.cpp.

◆ getNewFromOriginal() [3/4]

Operation * mlir::enzyme::MGradientUtils::getNewFromOriginal ( Operation * originst ) const

Definition at line 86 of file GradientUtils.cpp.

◆ getNewFromOriginal() [4/4]

SmallVector< mlir::Value, 1 > mlir::enzyme::MGradientUtils::getNewFromOriginal ( ValueRange originst ) const

Definition at line 66 of file GradientUtils.cpp.

Referenced by mlir::enzyme::detail::allocationForwardHandler(), mlir::enzyme::detail::branchingForwardHandler(), AutoDiffCallRev::cacheValues(), mlir::enzyme::detail::controlFlowForwardHandler(), AutoDiffCallFwd::createForwardModeTangent(), mlir::enzyme::MEnzymeLogic::CreateReverseDiff(), createTerminator(), mlir::enzyme::MEnzymeLogic::differentiate(), mlir::enzyme::MEnzymeLogic::handlePredecessors(), mlir::enzyme::detail::memoryIdentityForwardHandler(), mlir::enzyme::detail::regionTerminatorForwardHandler(), replaceOrigOpWith(), mlir::enzyme::MEnzymeLogic::visitChild(), and visitChild().

◆ getShadowType()

mlir::Type mlir::enzyme::MGradientUtils::getShadowType ( mlir::Type T )

inline

Definition at line 93 of file GradientUtils.h.

Referenced by GradientUtils::GradientUtils().

◆ invertPointerM()

mlir::Value mlir::enzyme::MGradientUtils::invertPointerM	(	mlir::Value	v,
		OpBuilder &	Builder2 )

Definition at line 117 of file GradientUtils.cpp.

Referenced by mlir::enzyme::detail::branchingForwardHandler(), mlir::enzyme::detail::controlFlowForwardHandler(), AutoDiffCallFwd::createForwardModeTangent(), AutoDiffCallRev::createReverseModeAdjoint(), createTerminator(), mlir::enzyme::detail::memoryIdentityForwardHandler(), and mlir::enzyme::detail::regionTerminatorForwardHandler().

◆ isConstantInstruction()

bool mlir::enzyme::MGradientUtils::isConstantInstruction ( mlir::Operation * v ) const

Definition at line 110 of file GradientUtils.cpp.

Referenced by isFullyInactive().

◆ isConstantValue()

bool mlir::enzyme::MGradientUtils::isConstantValue ( mlir::Value v ) const

Definition at line 113 of file GradientUtils.cpp.

Referenced by mlir::enzyme::detail::branchingForwardHandler(), mlir::enzyme::detail::controlFlowForwardHandler(), mlir::enzyme::detail::controlFlowForwardHandler(), AutoDiffCallFwd::createForwardModeTangent(), AutoDiffCallRev::createReverseModeAdjoint(), createTerminator(), mlir::enzyme::MEnzymeLogic::handlePredecessors(), mlir::enzyme::localizeGradients(), mlir::enzyme::detail::memoryIdentityForwardHandler(), mlir::enzyme::detail::regionTerminatorForwardHandler(), and mlir::enzyme::detail::returnReverseHandler().

◆ reindex_arguments()

static llvm::SmallVector< mlir::Value, 1 > mlir::enzyme::MGradientUtils::reindex_arguments	(	llvm::ArrayRef< mlir::Value >	vals,
		mlir::OperandRange	range )