EnzymeLogicReverse.cpp File Reference

#include "Dialect/Ops.h"
#include "Interfaces/AutoDiffOpInterface.h"
#include "Interfaces/AutoDiffTypeInterface.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/SymbolTable.h"
#include "mlir/Interfaces/FunctionInterfaces.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/ControlFlow/IR/ControlFlowOps.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Pass/PassRegistry.h"
#include "EnzymeLogic.h"
#include "Interfaces/GradientUtils.h"
#include "Interfaces/GradientUtilsReverse.h"
#include "llvm/ADT/ScopeExit.h"

Include dependency graph for EnzymeLogicReverse.cpp:

Go to the source code of this file.

Functions
void	handleReturns (Block *oBB, Block *newBB, Block *reverseBB, MGradientUtilsReverse *gutils)
static bool	isFullyInactive (Operation *op, MGradientUtils *gutils)

Function Documentation

handleReturns()

void handleReturns	(	Block *	oBB,
		Block *	newBB,
		Block *	reverseBB,
		MGradientUtilsReverse *	gutils )

Definition at line 25 of file EnzymeLogicReverse.cpp.

References mlir::enzyme::MGradientUtils::erase(), and mlir::enzyme::MGradientUtils::originalToNewFnOps.

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

isFullyInactive()

static bool isFullyInactive ( Operation * op, MGradientUtils * gutils )

static

Definition at line 43 of file EnzymeLogicReverse.cpp.

References mlir::enzyme::MGradientUtils::isConstantInstruction().

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