Enzyme main
Loading...
Searching...
No Matches
GradientUtils::ShadowRematerializer Struct Reference

#include "MLIR/Interfaces/GradientUtils.h"

Collaboration diagram for GradientUtils::ShadowRematerializer:

Public Member Functions

 ShadowRematerializer ()
 
 ShadowRematerializer (const llvm::SmallPtrSetImpl< llvm::Instruction * > &stores, const llvm::SmallPtrSetImpl< llvm::Instruction * > &frees, bool primalInitialize, llvm::Loop *LI)
 

Public Attributes

llvm::SmallPtrSet< llvm::Instruction *, 1 > stores
 Operations which must be rerun to rematerialize the original value.
 
llvm::SmallPtrSet< llvm::Instruction *, 1 > frees
 Operations which deallocate the value.
 
bool primalInitialize
 Whether the shadow must be initialized in the primal.
 
llvm::Loop * LI
 Loop scope (null if not loop scoped).
 

Detailed Description

Definition at line 267 of file GradientUtils.h.

Constructor & Destructor Documentation

◆ ShadowRematerializer() [1/2]

GradientUtils::ShadowRematerializer::ShadowRematerializer ( )
inline

Definition at line 281 of file GradientUtils.h.

◆ ShadowRematerializer() [2/2]

GradientUtils::ShadowRematerializer::ShadowRematerializer ( const llvm::SmallPtrSetImpl< llvm::Instruction * > & stores,
const llvm::SmallPtrSetImpl< llvm::Instruction * > & frees,
bool primalInitialize,
llvm::Loop * LI )
inline

Definition at line 283 of file GradientUtils.h.

Member Data Documentation

◆ frees

llvm::SmallPtrSet<llvm::Instruction *, 1> GradientUtils::ShadowRematerializer::frees

Operations which deallocate the value.

Definition at line 273 of file GradientUtils.h.

◆ LI

llvm::Loop* GradientUtils::ShadowRematerializer::LI

Loop scope (null if not loop scoped).

Definition at line 279 of file GradientUtils.h.

◆ primalInitialize

bool GradientUtils::ShadowRematerializer::primalInitialize

Whether the shadow must be initialized in the primal.

Definition at line 276 of file GradientUtils.h.

◆ stores

llvm::SmallPtrSet<llvm::Instruction *, 1> GradientUtils::ShadowRematerializer::stores

Operations which must be rerun to rematerialize the original value.

Definition at line 270 of file GradientUtils.h.


The documentation for this struct was generated from the following file: