Passes.h

Go to the documentation of this file.

//===- Passes.h - Enzyme pass include header  -----------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef ENZYME_PASSES_H
#define ENZYME_PASSES_H
 
#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"
 
namespace mlir {
class PatternRewriter;
class RewritePatternSet;
class DominanceInfo;
} // namespace mlir
 
namespace mlir {
// Forward declaration from Dialect.h
template <typename ConcreteDialect>
void registerDialect(DialectRegistry &registry);
 
namespace arith {
class ArithDialect;
} // end namespace arith
 
namespace complex {
class ComplexDialect;
} // end namespace complex
 
namespace cf {
class ControlFlowDialect;
} // end namespace cf
 
namespace scf {
class SCFDialect;
} // end namespace scf
 
namespace memref {
class MemRefDialect;
} // end namespace memref
 
namespace func {
class FuncDialect;
} // end namespace func
 
namespace affine {
class AffineDialect;
} // end namespace affine
 
namespace tensor {
class TensorDialect;
} // end namespace tensor
 
namespace LLVM {
class LLVMDialect;
} // end namespace LLVM
 
namespace tensor {
class TensorDialect;
} // end namespace tensor
 
namespace linalg {
class LinalgDialect;
} // end namespace linalg
 
namespace memref {
class MemRefDialect;
} // end namespace memref
 
namespace enzyme {
 
#define GEN_PASS_DECL
#include "Passes/Passes.h.inc"
 
#define GEN_PASS_REGISTRATION
#include "Passes/Passes.h.inc"
 
class AutoDiffOp;
bool inlineAutodiffOp(enzyme::AutoDiffOp &op, RewriterBase &rewriter,
                      SymbolTableCollection &symbolTable);
 
/// 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.
/// `maxHoistedBytes` bounds the size of a hoisted buffer; 0 means no bound.
/// Returns whether anything moved.
bool hoistLoopAllocations(LoopLikeOpInterface loop,
                          uint64_t maxHoistedBytes = 0);
} // namespace enzyme
 
} // end namespace mlir
 
#endif // ENZYME_PASSES_H