#include "MLIR/Interfaces/EnzymeLogic.h"

Collaboration diagram for mlir::enzyme::MEnzymeLogic:

[legend]

Classes
struct	MForwardCacheKey

struct	MReverseCacheKey

Public Member Functions
FunctionOpInterface	CreateForwardDiff (FunctionOpInterface fn, std::vector< DIFFE_TYPE > retType, std::vector< DIFFE_TYPE > constants, MTypeAnalysis &TA, std::vector< bool > returnPrimals, DerivativeMode mode, bool freeMemory, size_t width, mlir::Type addedType, MFnTypeInfo type_args, std::vector< bool > volatile_args, void *augmented, bool omp, llvm::StringRef postpasses, bool verifyPostPasses, bool strongZero)

FunctionOpInterface	CreateReverseDiff (FunctionOpInterface fn, std::vector< DIFFE_TYPE > retType, std::vector< DIFFE_TYPE > constants, MTypeAnalysis &TA, std::vector< bool > returnPrimals, std::vector< bool > returnShadows, DerivativeMode mode, bool freeMemory, size_t width, mlir::Type addedType, MFnTypeInfo type_args, std::vector< bool > volatile_args, void *augmented, bool omp, llvm::StringRef postpasses, bool verifyPostPasses, bool strongZero)

void	initializeShadowValues (SmallVector< mlir::Block * > &dominatorToposortBlocks, MGradientUtilsReverse *gutils)

void	handlePredecessors (Block oBB, Block newBB, Block reverseBB, MGradientUtilsReverse gutils, llvm::function_ref< buildReturnFunction > buildReturnOp)

LogicalResult	visitChildren (Block oBB, Block reverseBB, MGradientUtilsReverse *gutils)

LogicalResult	visitChild (Operation op, OpBuilder &builder, MGradientUtilsReverse gutils)

void	mapInvertArguments (Block oBB, Block reverseBB, MGradientUtilsReverse *gutils)

LogicalResult	differentiate (MGradientUtilsReverse *gutils, Region &oldRegion, Region &newRegion, llvm::function_ref< buildReturnFunction > buildFuncRetrunOp, std::function< std::pair< Value, Value >(Type)> cacheCreator)

Public Attributes
std::map< MForwardCacheKey, FunctionOpInterface >	ForwardCachedFunctions

std::map< MReverseCacheKey, FunctionOpInterface >	ReverseCachedFunctions

Detailed Description

Definition at line 46 of file EnzymeLogic.h.

Member Function Documentation

◆ CreateForwardDiff()

FunctionOpInterface mlir::enzyme::MEnzymeLogic::CreateForwardDiff	(	FunctionOpInterface	fn,
		std::vector< DIFFE_TYPE >	retType,
		std::vector< DIFFE_TYPE >	constants,
		MTypeAnalysis &	TA,
		std::vector< bool >	returnPrimals,
		DerivativeMode	mode,
		bool	freeMemory,
		size_t	width,
		mlir::Type	addedType,
		MFnTypeInfo	type_args,
		std::vector< bool >	volatile_args,
		void *	augmented,
		bool	omp,
		llvm::StringRef	postpasses,
		bool	verifyPostPasses,
		bool	strongZero )

Definition at line 78 of file EnzymeLogic.cpp.

References ForwardCachedFunctions.

Referenced by AutoDiffCallFwd::createForwardModeTangent().

◆ CreateReverseDiff()

FunctionOpInterface MEnzymeLogic::CreateReverseDiff	(	FunctionOpInterface	fn,
		std::vector< DIFFE_TYPE >	retType,
		std::vector< DIFFE_TYPE >	constants,
		MTypeAnalysis &	TA,
		std::vector< bool >	returnPrimals,
		std::vector< bool >	returnShadows,
		DerivativeMode	mode,
		bool	freeMemory,
		size_t	width,
		mlir::Type	addedType,
		MFnTypeInfo	type_args,
		std::vector< bool >	volatile_args,
		void *	augmented,
		bool	omp,
		llvm::StringRef	postpasses,
		bool	verifyPostPasses,
		bool	strongZero )

Definition at line 192 of file EnzymeLogicReverse.cpp.

References mlir::enzyme::MGradientUtilsReverse::CreateFromClone(), mlir::enzyme::MDiffeGradientUtils::diffe(), differentiate(), mlir::enzyme::MGradientUtils::forceAugmentedReturns(), mlir::enzyme::MGradientUtils::getNewFromOriginal(), mlir::enzyme::MGradientUtils::newFunc, mlir::enzyme::MGradientUtils::oldFunc, OUT_DIFF, and ReverseCachedFunctions.

Referenced by AutoDiffCallRev::createReverseModeAdjoint().

◆ differentiate()

LogicalResult MEnzymeLogic::differentiate	(	MGradientUtilsReverse *	gutils,
		Region &	oldRegion,
		Region &	newRegion,
		llvm::function_ref< buildReturnFunction >	buildFuncRetrunOp,
		std::function< std::pair< Value, Value >(Type)>	cacheCreator )

Definition at line 171 of file EnzymeLogicReverse.cpp.

References mlir::enzyme::MGradientUtilsReverse::createReverseModeBlocks(), mlir::enzyme::MGradientUtilsReverse::deregisterCacheCreatorHook(), mlir::enzyme::MGradientUtils::getNewFromOriginal(), handlePredecessors(), handleReturns(), mlir::enzyme::MGradientUtilsReverse::mapReverseModeBlocks, mlir::enzyme::MGradientUtilsReverse::registerCacheCreatorHook(), and visitChildren().

Referenced by CreateReverseDiff().

◆ handlePredecessors()

void MEnzymeLogic::handlePredecessors	(	Block *	oBB,
		Block *	newBB,
		Block *	reverseBB,
		MGradientUtilsReverse *	gutils,
		llvm::function_ref< buildReturnFunction >	buildReturnOp )