Enzyme.h

Go to the documentation of this file.

//===- Enzyme.h - Automatic Differentiation Transformation Pass    -------===//
//
//                             Enzyme Project
//
// Part of the Enzyme 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
//
// If using this code in an academic setting, please cite the following:
// @incollection{enzymeNeurips,
// title = {Instead of Rewriting Foreign Code for Machine Learning,
//          Automatically Synthesize Fast Gradients},
// author = {Moses, William S. and Churavy, Valentin},
// booktitle = {Advances in Neural Information Processing Systems 33},
// year = {2020},
// note = {To appear in},
// }
//
//===----------------------------------------------------------------------===//
//
// This file declares Enzyme, a transformation pass that takes replaces calls
// to function calls to *__enzyme_autodiff* with a call to the derivative of
// the function passed as the first argument.
//
//===----------------------------------------------------------------------===//
 
#ifndef ENZYME_H
#define ENZYME_H
 
#include "llvm/Pass.h"
#include "llvm/Passes/PassBuilder.h"
 
llvm::ModulePass *createEnzymePass(bool PostOpt = false);
void augmentPassBuilder(llvm::PassBuilder &PB);
 
#endif // ENZYME_H