CoreDialectsAutoDiffImplementations.h File Reference

#include "Interfaces/AutoDiffOpInterface.h"
#include "mlir/Support/LogicalResult.h"
#include "llvm/ADT/DenseSet.h"

Include dependency graph for CoreDialectsAutoDiffImplementations.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	mlir::enzyme::detail::AutoDiffUsingControlFlow< OpTy >
class	mlir::enzyme::detail::AutoDiffUsingBranch< OpTy >
class	mlir::enzyme::detail::AutoDiffUsingRegionTerminator< OpTy >
class	mlir::enzyme::detail::NoopRevAutoDiffInterface< OpTy >
class	mlir::enzyme::detail::ReturnRevAutoDiffInterface< OpTy >
class	mlir::enzyme::detail::AutoDiffUsingMemoryIdentity< OpTy, storedvals >
class	mlir::enzyme::detail::AutoDiffUsingAllocationFwd< OpTy >
class	mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >

Namespaces
namespace	mlir
namespace	mlir::enzyme
namespace	mlir::enzyme::detail

Functions
LogicalResult	mlir::enzyme::detail::controlFlowForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils)
LogicalResult	mlir::enzyme::detail::controlFlowForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils, const llvm::SmallDenseSet< unsigned > &operandPositionsToShadow, const llvm::SmallDenseSet< unsigned > &resultPositionsToShadow)
void	mlir::enzyme::detail::branchingForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils)
void	mlir::enzyme::detail::regionTerminatorForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils)
void	mlir::enzyme::detail::returnReverseHandler (Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils)
LogicalResult	mlir::enzyme::detail::memoryIdentityForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils, ArrayRef< int > storedVals)
LogicalResult	mlir::enzyme::detail::allocationForwardHandler (Operation *op, OpBuilder &builder, MGradientUtils *gutils, bool zero)
template<typename OpTy >
void	mlir::enzyme::registerAutoDiffUsingControlFlowInterface (MLIRContext &context)
template<typename OpTy >
void	mlir::enzyme::registerAutoDiffUsingBranchInterface (MLIRContext &context)
template<typename OpTy >
void	mlir::enzyme::registerAutoDiffUsingRegionTerminatorInterface (MLIRContext &context)
template<typename OpTy >
void	mlir::enzyme::registerAutoDiffUsingReturnInterface (MLIRContext &context)
template<typename OpTy , int... storedvals>
void	mlir::enzyme::registerAutoDiffUsingMemoryIdentityInterface (MLIRContext &context)
template<typename OpTy >
void	mlir::enzyme::registerAutoDiffUsingAllocationInterface (MLIRContext &context)
void	mlir::enzyme::registerAffineDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerArithDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerBuiltinDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerLLVMDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerLLVMExtDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerNVVMDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerMemRefDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerComplexDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerSCFDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerCFDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerLinalgDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerMathDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerFuncDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerTensorDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerEnzymeDialectAutoDiffInterface (DialectRegistry &registry)
void	mlir::enzyme::registerCoreDialectAutodiffInterfaces (DialectRegistry &registry)
mlir::TypedAttr	mlir::enzyme::getConstantAttr (mlir::Type type, llvm::StringRef value)