AutoDiffOpInterface.h

Go to the documentation of this file.

//===- AutoDiffOpInterface.h - Op interface for auto diff- -------* C++ -*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This file defines the interfaces necessary to implement scalable automatic
// differentiation across an unbounded number of MLIR IR constructs.
//
//===----------------------------------------------------------------------===//
 
#ifndef ENZYME_MLIR_INTERFACES_AUTODIFFOPINTERFACE_H
#define ENZYME_MLIR_INTERFACES_AUTODIFFOPINTERFACE_H
 
#include "mlir/IR/OpDefinition.h"
#include "mlir/Interfaces/ControlFlowInterfaces.h"
 
namespace mlir {
 
class OpBuilder;
class Operation;
class IRMapping;
class PatternRewriter;
 
namespace enzyme {
 
class MGradientUtils;
class MGradientUtilsReverse;
 
}; // namespace enzyme
}; // namespace mlir
 
#include "MLIR/Interfaces/AutoDiffOpInterface.h.inc"
 
namespace mlir {
namespace enzyme {
inline bool isFunctionReturnLike(Operation *op) {
  return op->hasTrait<OpTrait::ReturnLike>() ||
         isa<FunctionReturnOpInterface>(op);
}
} // namespace enzyme
} // namespace mlir
 
#endif // ENZYME_MLIR_INTERFACES_AUTODIFFOPINTERFACE_H