| allocCloneSlots(OpBuilder &b, Location loc, int64_t budget, Value proto, ClonableTypeInterface iface) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| binomialCheckpointingAttrName() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| cacheBinomial(OpName forOp, int64_t budget, MGradientUtilsReverse *gutils) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| CacheCreator typedef | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | |
| cachePeriodic(OpName forOp, Operation *op, MGradientUtilsReverse *gutils) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| castToType(OpBuilder &builder, Location loc, Value v, Type targetType) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| checkpointBufferType(int64_t budget, Type t) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| checkpointingEnabled(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| checkpointPeriodAttrName() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| checkpointRow(OpBuilder &b, Location loc, Value buf, Value slot, MemRefType rowTy) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| cloneSlot(OpBuilder &builder, Location loc, Value base, Value buf, Value slot, int64_t budget) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| copyBlockWithoutTerminator(OpBuilder &builder, Block *b, MGradientUtilsReverse *gutils, IRMapping &mapping) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| createScaffoldForLoop(OpBuilder &b, Location loc, Value lb, Value ub, Value step, ValueRange inits) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| createScaffoldWhileLoop(OpBuilder &b, Location loc, ValueRange inits, llvm::function_ref< Value(OpBuilder &, Location, ValueRange)> cond) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| createStore(OpBuilder &b, Location loc, int64_t budget, Type valTy) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| destroyStore(OpBuilder &b, Location loc, Value store) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitAdd(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitCmpEQ(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitCmpLT(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitConst(OpBuilder &b, Location loc, int64_t v) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitDivU(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitMin(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitMul(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitSelect(OpBuilder &b, Location loc, Value c, Value t, Value f) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| emitSub(OpBuilder &b, Location loc, Value l, Value r) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| enableCheckpointingAttrName() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| finalizeScaffoldLoop(OpBuilder &b, Location loc, const ScaffoldLoop &loop, ValueRange yields) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| freeCloneSlots(OpBuilder &builder, Value ref, Value buf, int64_t budget) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getBodyBlock(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getCarriedResults(Operation *loop) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getCarriedTerminatorOperands(Operation *term) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getCheckpointBudget(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getIndexLikeType(OpBuilder &builder) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getInductionVar(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getInits(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getNumIterationsValue(OpBuilder &builder, Location loc, OpName forOp, MGradientUtilsReverse *gutils) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getNumRegionIterArgs(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getPrimalResults(OpBuilder &b, Location loc, OpName forOp, ValueRange state, MGradientUtilsReverse *gutils) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| getStaticPeriodicSchedule(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| hasBinomialAttr(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| hoistPlaceholdersBefore(Operation *root, Operation *before) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| isDefinedInBlock(Block *b, Value v) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| loadCheckpoint(OpBuilder &b, Location loc, Value buf, Value slot, Type valTy) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| loadSlot(OpBuilder &b, Location loc, Value store, Value slot, Type valTy) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| makeSegmentCacheCreator(MGradientUtilsReverse *gutils, Operation *anchor) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| materializeDynamicSchedule(OpBuilder &builder, Location loc, OpName forOp, MGradientUtilsReverse *gutils, PeriodicSchedule &sched) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| materializeSegmentValues(OpBuilder &builder, Location loc, PeriodicSchedule &sched) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| needsBinomialCheckpointing(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| needsCheckpointing(OpName forOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| preserveAttributesButCheckpointing(Operation *newOp, Operation *oldOp) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| primeSegmentGradients(OpBuilder &builder, MGradientUtilsReverse *gutils, Block *origBody, Block *revLoopBody, ArrayRef< bool > operandsActive) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| primeStepGradients(OpBuilder &builder, MGradientUtilsReverse *gutils, Block *origBody, Block *revOuterBody, ArrayRef< bool > operandsActive) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| publishesCopiedSeeds() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| reverseBinomial(OpName forOp, int64_t budget, OpBuilder &builder, MGradientUtilsReverse *gutils, SmallVector< Value > caches, ArrayRef< bool > operandsActive, ArrayRef< Value > incomingGradients) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| reversePeriodic(OpName forOp, Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils, SmallVector< Value > caches, ArrayRef< bool > operandsActive, ArrayRef< Value > incomingGradients) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| segmentBase(OpBuilder &builder, Location loc, const PeriodicSchedule &sched, Value index) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| segmentLength(OpBuilder &builder, Location loc, const PeriodicSchedule &sched, Value base) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| setInsertionPointToBodyEnd(OpBuilder &builder, Block *block) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| splitOutsideRefs(OpName forOp, SmallVectorImpl< Value > &mutableRefs, SmallVectorImpl< Value > &immutableRefs) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| storeCheckpoint(OpBuilder &b, Location loc, Value buf, Value slot, Value val) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| storesAreLoopCarried() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| storeSlot(OpBuilder &b, Location loc, Value store, Value slot, Value val) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| supportsDynamicPeriodic() | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| tryCacheValues(OpName forOp, Operation *op, MGradientUtilsReverse *gutils) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
| tryCreateReverseModeAdjoint(OpName forOp, Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils, SmallVector< Value > caches, ArrayRef< bool > operandsActive, ArrayRef< Value > incomingGradients) | mlir::enzyme::LoopCheckpointing< FinalClass, OpName > | inlinestatic |
1.12.0