Enzyme main
Loading...
Searching...
No Matches
LoopCheckpointing.h File Reference
#include "Dialect/Ops.h"
#include "Interfaces/AutoDiffTypeInterface.h"
#include "Interfaces/GradientUtilsReverse.h"
#include "Passes/RemovalUtils.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/IR/IRMapping.h"
#include "mlir/Transforms/RegionUtils.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/ScopeExit.h"
#include <array>
#include <cmath>
#include <functional>
#include <optional>
#include <utility>
Include dependency graph for LoopCheckpointing.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mlir::enzyme::LoopCheckpointing< FinalClass, OpName >
 
struct  mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::PeriodicSchedule
 
struct  mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout
 
struct  mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::ScaffoldLoop
 
class  mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::SegmentCacheCreatorGuard
 

Namespaces

namespace  mlir
 
namespace  mlir::enzyme