|
Enzyme main
|
#include "Ops.h"#include "Interfaces/AutoDiffTypeInterface.h"#include "mlir/Dialect/LLVMIR/LLVMTypes.h"#include "mlir/IR/AffineExpr.h"#include "mlir/IR/Builders.h"#include "mlir/IR/Matchers.h"#include "mlir/IR/PatternMatch.h"#include "mlir/IR/Value.h"#include "mlir/Interfaces/MemorySlotInterfaces.h"#include "mlir/Dialect/Func/IR/FuncOps.h"#include "mlir/Dialect/MemRef/IR/MemRef.h"#include "mlir/IR/IntegerSet.h"#include "llvm/ADT/STLExtras.h"#include "llvm/ADT/SmallVector.h"#include "llvm/Support/ErrorHandling.h"#include "llvm/Support/LogicalResult.h"#include <type_traits>
Go to the source code of this file.
Classes | |
| struct | EnzymeOpCreator< AutoDiffOp > |
| struct | EnzymeOpCreator< AutoDiffRegionOp > |
| struct | EnzymeOpCreator< ForwardDiffOp > |
| struct | EnzymeOpCreator< ForwardDiffRegionOp > |
| class | FwdInpOpt< SourceOp > |
| Modifies input activites for the FwdDiffOp The activity promotion flow is as follows (depending on variable use): More... | |
| class | FwdRetOpt< SourceOp > |
| Modifies return activites for the FwdDiffOp The activity promotion flow is as follows (depending on variable use): More... | |
| class | ReverseRetOpt< SourceOp > |
| Modifies activities for the AutoDiffOp. More... | |
| class | RemoveUnusedArgs< SourceRegionOp > |
Macros | |
| #define | DEBUG_TYPE "enzyme" |
Functions | |
| static bool | isMutable (Type type) |
|
inlinestatic |
Definition at line 235 of file Ops.cpp.
Referenced by FwdInpOpt< SourceOp >::matchAndRewrite(), FwdRetOpt< SourceOp >::matchAndRewrite(), and ReverseRetOpt< SourceOp >::matchAndRewrite().