AutoDiffCallRev Class Reference

Inheritance diagram for AutoDiffCallRev:

Collaboration diagram for AutoDiffCallRev:

Public Member Functions
LogicalResult	createReverseModeAdjoint (Operation *orig, OpBuilder &builder, MGradientUtilsReverse *gutils, SmallVector< Value > caches) const
SmallVector< Value >	cacheValues (Operation *orig, MGradientUtilsReverse *gutils) const
void	createShadowValues (Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils) const

Detailed Description

Definition at line 117 of file FuncAutoDiffOpInterfaceImpl.cpp.

Member Function Documentation

cacheValues()

SmallVector< Value > AutoDiffCallRev::cacheValues ( Operation * orig, MGradientUtilsReverse * gutils ) const

inline

Definition at line 216 of file FuncAutoDiffOpInterfaceImpl.cpp.

References mlir::enzyme::MGradientUtils::getNewFromOriginal(), and mlir::enzyme::MGradientUtilsReverse::initAndPushCache().

createReverseModeAdjoint()

LogicalResult AutoDiffCallRev::createReverseModeAdjoint ( Operation * orig, OpBuilder & builder, MGradientUtilsReverse * gutils, SmallVector< Value > caches ) const

inline

Definition at line 121 of file FuncAutoDiffOpInterfaceImpl.cpp.

References mlir::enzyme::MGradientUtilsReverse::addToDiffe(), CONSTANT, mlir::enzyme::MEnzymeLogic::CreateReverseDiff(), mlir::enzyme::MDiffeGradientUtils::diffe(), DUP_ARG, mlir::enzyme::MTypeAnalysis::getAnalyzedTypeInfo(), mlir::enzyme::MGradientUtils::invertPointerM(), mlir::enzyme::MGradientUtils::isConstantValue(), mlir::enzyme::MGradientUtils::Logic, mlir::enzyme::MGradientUtils::omp, OUT_DIFF, mlir::enzyme::MGradientUtilsReverse::popCache(), mlir::enzyme::MGradientUtils::postpasses, ReverseModeGradient, mlir::enzyme::MGradientUtils::strongZero, mlir::enzyme::MGradientUtils::TA, mlir::enzyme::MGradientUtils::verifyPostPasses, and mlir::enzyme::MGradientUtils::width.

createShadowValues()

void AutoDiffCallRev::createShadowValues ( Operation * op, OpBuilder & builder, MGradientUtilsReverse * gutils ) const

inline

Definition at line 235 of file FuncAutoDiffOpInterfaceImpl.cpp.

---

The documentation for this class was generated from the following file:

• MLIR/Implementations/FuncAutoDiffOpInterfaceImpl.cpp