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

#include "MLIR/Interfaces/GradientUtils.h"

Collaboration diagram for GradientUtils::Rematerializer:

Public Member Functions

 Rematerializer ()
 
 Rematerializer (llvm::ArrayRef< llvm::LoadInst * > loads, llvm::ArrayRef< LoadLikeCall > loadLikeCalls, const llvm::SmallPtrSetImpl< llvm::Instruction * > &stores, const llvm::SmallPtrSetImpl< llvm::Instruction * > &frees, llvm::Loop *LI, llvm::CallInst *nonRepeatableWritingCall)
 

Public Attributes

llvm::SmallVector< llvm::LoadInst *, 1 > loads
 
llvm::SmallVector< LoadLikeCall, 1 > loadLikeCalls
 
llvm::SmallPtrSet< llvm::Instruction *, 1 > stores
 
llvm::SmallPtrSet< llvm::Instruction *, 1 > frees
 
llvm::Loop * LI
 
llvm::CallInst * nonRepeatableWritingCall
 

Detailed Description

Definition at line 231 of file GradientUtils.h.

Constructor & Destructor Documentation

◆ Rematerializer() [1/2]

GradientUtils::Rematerializer::Rematerializer ( )
inline

Definition at line 254 of file GradientUtils.h.

◆ Rematerializer() [2/2]

GradientUtils::Rematerializer::Rematerializer ( llvm::ArrayRef< llvm::LoadInst * > loads,
llvm::ArrayRef< LoadLikeCall > loadLikeCalls,
const llvm::SmallPtrSetImpl< llvm::Instruction * > & stores,
const llvm::SmallPtrSetImpl< llvm::Instruction * > & frees,
llvm::Loop * LI,
llvm::CallInst * nonRepeatableWritingCall )
inline

Definition at line 255 of file GradientUtils.h.

Member Data Documentation

◆ frees

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

Definition at line 243 of file GradientUtils.h.

◆ LI

llvm::Loop* GradientUtils::Rematerializer::LI

Definition at line 246 of file GradientUtils.h.

◆ loadLikeCalls

llvm::SmallVector<LoadLikeCall, 1> GradientUtils::Rematerializer::loadLikeCalls

Definition at line 236 of file GradientUtils.h.

◆ loads

llvm::SmallVector<llvm::LoadInst *, 1> GradientUtils::Rematerializer::loads

Definition at line 233 of file GradientUtils.h.

◆ nonRepeatableWritingCall

llvm::CallInst* GradientUtils::Rematerializer::nonRepeatableWritingCall

Definition at line 252 of file GradientUtils.h.

◆ stores

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

Definition at line 240 of file GradientUtils.h.


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