Enzyme main
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout Struct Reference

#include "MLIR/Implementations/LoopCheckpointing.h"

Collaboration diagram for mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout:
Collaboration graph
[legend]

Public Member Functions

size_t ckptBuf (size_t i) const
 
size_t idxBuf () const
 
size_t mutableBegin () const
 
size_t mutableWidth () const
 
size_t cloneBuf (size_t r) const
 
size_t shadow (size_t r) const
 
size_t immutable (size_t i) const
 
size_t numIters () const
 
size_t start () const
 
size_t step () const
 
size_t size () const
 

Static Public Member Functions

static BinomialCacheLayout get (OpName forOp, ArrayRef< Value > mutableRefs, ArrayRef< Value > immutableRefs, bool isDynamic, MGradientUtilsReverse *gutils)
 

Public Attributes

size_t numIterArgs = 0
 
SmallVector< bool > mutableActive
 
size_t numImmutable = 0
 
bool isDynamic = false
 

Detailed Description

template<typename FinalClass, typename OpName>
struct mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout

Definition at line 389 of file LoopCheckpointing.h.

Member Function Documentation

◆ ckptBuf()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::ckptBuf ( size_t i) const
inline

Definition at line 408 of file LoopCheckpointing.h.

◆ cloneBuf()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::cloneBuf ( size_t r) const
inline

Definition at line 418 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableActive, and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableBegin().

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::shadow().

◆ get()

template<typename FinalClass , typename OpName >
static BinomialCacheLayout mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::get ( OpName forOp,
ArrayRef< Value > mutableRefs,
ArrayRef< Value > immutableRefs,
bool isDynamic,
MGradientUtilsReverse * gutils )
inlinestatic

Definition at line 395 of file LoopCheckpointing.h.

References mlir::enzyme::MGradientUtils::isConstantValue(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::isDynamic, mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableActive, mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numImmutable, and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIterArgs.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::cacheBinomial(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::reverseBinomial().

◆ idxBuf()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::idxBuf ( ) const
inline

Definition at line 409 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIterArgs.

◆ immutable()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::immutable ( size_t i) const
inline

Definition at line 428 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableBegin(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableWidth().

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::size().

◆ mutableBegin()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableBegin ( ) const
inline

Definition at line 411 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIterArgs.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::cloneBuf(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::immutable().

◆ mutableWidth()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableWidth ( ) const
inline

Definition at line 412 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableActive.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::immutable().

◆ numIters()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters ( ) const
inline

Definition at line 431 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::immutable(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::isDynamic, and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numImmutable.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::start(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::step().

◆ shadow()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::shadow ( size_t r) const
inline

Definition at line 424 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::cloneBuf(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableActive.

◆ size()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::size ( ) const
inline

Definition at line 437 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::immutable(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::isDynamic, and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numImmutable.

◆ start()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::start ( ) const
inline

Definition at line 435 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters().

◆ step()

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::step ( ) const
inline

Definition at line 436 of file LoopCheckpointing.h.

References mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters().

Member Data Documentation

◆ isDynamic

template<typename FinalClass , typename OpName >
bool mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::isDynamic = false

Definition at line 393 of file LoopCheckpointing.h.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::get(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::size().

◆ mutableActive

template<typename FinalClass , typename OpName >
SmallVector<bool> mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableActive

Definition at line 391 of file LoopCheckpointing.h.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::cloneBuf(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::get(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableWidth(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::shadow().

◆ numImmutable

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numImmutable = 0

Definition at line 392 of file LoopCheckpointing.h.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::get(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIters(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::size().

◆ numIterArgs

template<typename FinalClass , typename OpName >
size_t mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::numIterArgs = 0

Definition at line 390 of file LoopCheckpointing.h.

Referenced by mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::get(), mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::idxBuf(), and mlir::enzyme::LoopCheckpointing< FinalClass, OpName >::BinomialCacheLayout::mutableBegin().

The documentation for this struct was generated from the following file:
Generated on Tue Oct 13 2026 01:33:29 for Enzyme by doxygen 1.12.0