|
Enzyme main
|
#include "MLIR/Interfaces/GradientUtils.h"

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). | |
Definition at line 267 of file GradientUtils.h.
|
inline |
Definition at line 281 of file GradientUtils.h.
|
inline |
Definition at line 283 of file GradientUtils.h.
| llvm::SmallPtrSet<llvm::Instruction *, 1> GradientUtils::ShadowRematerializer::frees |
Operations which deallocate the value.
Definition at line 273 of file GradientUtils.h.
| llvm::Loop* GradientUtils::ShadowRematerializer::LI |
Loop scope (null if not loop scoped).
Definition at line 279 of file GradientUtils.h.
| bool GradientUtils::ShadowRematerializer::primalInitialize |
Whether the shadow must be initialized in the primal.
Definition at line 276 of file GradientUtils.h.
| 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.