ActivityAnnotations.cpp File Reference

#include "ActivityAnnotations.h"
#include "DataFlowAliasAnalysis.h"
#include "DataFlowLattice.h"
#include "Dialect/Dialect.h"
#include "Dialect/Ops.h"
#include "Interfaces/AutoDiffOpInterface.h"
#include "mlir/Analysis/DataFlow/DenseAnalysis.h"
#include "mlir/Analysis/DataFlow/SparseAnalysis.h"
#include "mlir/Analysis/DataFlow/Utils.h"
#include "mlir/Analysis/DataFlowFramework.h"
#include "mlir/Interfaces/FunctionInterfaces.h"
#include "llvm/ADT/TypeSwitch.h"
#include "llvm/Support/raw_ostream.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"

Include dependency graph for ActivityAnnotations.cpp:

Go to the source code of this file.

Functions
static bool	isPossiblyActive (Type type)
void	deserializeReturnOrigins (ArrayAttr returnOrigins, SmallVectorImpl< enzyme::ValueOriginSet > &out)
template<typename KeyT , typename ElementT >
void	printMapOfSetsLattice (const DenseMap< KeyT, enzyme::SetLattice< ElementT > > map, raw_ostream &os)
std::optional< Value >	getStored (Operation *op)
std::optional< Value >	getCopySource (Operation *op)
static void	deserializePointsTo (ArrayAttr summaryAttr, DenseMap< DistinctAttr, enzyme::ValueOriginSet > &summaryMap)

Function Documentation

deserializePointsTo()

static void deserializePointsTo ( ArrayAttr summaryAttr, DenseMap< DistinctAttr, enzyme::ValueOriginSet > & summaryMap )

static

Definition at line 503 of file ActivityAnnotations.cpp.

References mlir::enzyme::SetLattice< OriginAttr >::getUndefined(), mlir::enzyme::undefinedSetString, and mlir::enzyme::unknownSetString.

Referenced by mlir::enzyme::DenseActivityAnnotationAnalysis::visitCallControlFlowTransfer(), mlir::enzyme::DenseBackwardActivityAnnotationAnalysis::visitCallControlFlowTransfer(), and mlir::enzyme::PointsToPointerAnalysis::visitCallControlFlowTransfer().

deserializeReturnOrigins()

void deserializeReturnOrigins	(	ArrayAttr	returnOrigins,
		SmallVectorImpl< enzyme::ValueOriginSet > &	out )

Definition at line 135 of file ActivityAnnotations.cpp.

References mlir::enzyme::SetLattice< ValueT >::insert(), and mlir::enzyme::SetLattice< ValueT >::markUnknown().

Referenced by mlir::enzyme::BackwardActivityAnnotationAnalysis::visitExternalCall(), and mlir::enzyme::ForwardActivityAnnotationAnalysis::visitExternalCall().

getCopySource()

std::optional< Value > getCopySource

(

Operation *

op

)

Definition at line 220 of file DataFlowActivityAnalysis.cpp.

Referenced by mlir::enzyme::DenseActivityAnnotationAnalysis::visitOperation(), mlir::enzyme::DenseBackwardActivityAnalysis::visitOperation(), mlir::enzyme::DenseBackwardActivityAnnotationAnalysis::visitOperation(), mlir::enzyme::DenseForwardActivityAnalysis::visitOperation(), and mlir::enzyme::PointsToPointerAnalysis::visitOperation().

getStored()

std::optional< Value > getStored

(

Operation *

op

)

Definition at line 204 of file DataFlowActivityAnalysis.cpp.

isPossiblyActive()

static bool isPossiblyActive ( Type type )

static

Definition at line 22 of file ActivityAnnotations.cpp.

References isPossiblyActive().

Referenced by isPossiblyActive(), mlir::enzyme::runActivityAnnotations(), mlir::enzyme::ForwardActivityAnnotationAnalysis::setToEntryState(), and mlir::enzyme::DenseActivityAnnotationAnalysis::visitOperation().

printMapOfSetsLattice()

template<typename KeyT , typename ElementT >

void printMapOfSetsLattice	(	const DenseMap< KeyT, enzyme::SetLattice< ElementT > >	map,
		raw_ostream &	os )

Definition at line 361 of file ActivityAnnotations.cpp.

Referenced by mlir::enzyme::BackwardOriginsMap::print(), and mlir::enzyme::ForwardOriginsMap::print().