#include "../../Utils.h"
#include "mlir/Conversion/LLVMCommon/LoweringOptions.h"
#include "mlir/Pass/Pass.h"
#include <memory>
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Complex/IR/Complex.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Interfaces/LoopLikeInterface.h"
#include "Dialect/Dialect.h"
#include "Passes/Passes.h.inc"
Go to the source code of this file.
|
| template<typename ConcreteDialect > |
| void | mlir::registerDialect (DialectRegistry ®istry) |
| |
| bool | mlir::enzyme::inlineAutodiffOp (enzyme::AutoDiffOp &op, RewriterBase &rewriter, SymbolTableCollection &symbolTable) |
| |
| bool | mlir::enzyme::hoistLoopAllocations (LoopLikeOpInterface loop, uint64_t maxHoistedBytes=0) |
| | Moves allocation/deallocation pairs that are private to one iteration of loop out of it, so the buffer is allocated once for the whole loop.
|
| |
◆ GEN_PASS_DECL
◆ GEN_PASS_REGISTRATION
| #define GEN_PASS_REGISTRATION |