MLIR  19.0.0git
Typedefs | Functions
OneShotModuleBufferize.cpp File Reference
#include "mlir/Dialect/Bufferization/Transforms/OneShotModuleBufferize.h"
#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Bufferization/Transforms/Bufferize.h"
#include "mlir/Dialect/Bufferization/Transforms/FuncBufferizableOpInterfaceImpl.h"
#include "mlir/Dialect/Bufferization/Transforms/OneShotAnalysis.h"
#include "mlir/Dialect/Bufferization/Transforms/Transforms.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/IR/Operation.h"

Go to the source code of this file.

Typedefs

using FuncCallerMap = DenseMap< func::FuncOp, DenseSet< Operation * > >
 A mapping of FuncOps to their callers. More...
 

Functions

static FuncAnalysisStategetOrCreateFuncAnalysisState (OneShotAnalysisState &state)
 Get or create FuncAnalysisState. More...
 
static func::ReturnOp getAssumedUniqueReturnOp (func::FuncOp funcOp)
 Return the unique ReturnOp that terminates funcOp. More...
 
static void removeBufferizationAttributes (BlockArgument bbArg)
 Remove bufferization attributes on FuncOp arguments. More...
 
static func::FuncOp getCalledFunction (func::CallOp callOp)
 Return the func::FuncOp called by callOp. More...
 
static void equivalenceAnalysis (func::FuncOp funcOp, OneShotAnalysisState &state, FuncAnalysisState &funcState)
 Gather equivalence info of CallOps. More...
 
static bool hasTensorSignature (func::FuncOp funcOp)
 Return "true" if the given function signature has tensor semantics. More...
 
static LogicalResult getFuncOpsOrderedByCalls (ModuleOp moduleOp, SmallVectorImpl< func::FuncOp > &orderedFuncOps, FuncCallerMap &callerMap)
 Store all functions of the moduleOp in orderedFuncOps, sorted by callee-caller order (i.e. More...
 
static void foldMemRefCasts (func::FuncOp funcOp)
 Fold return values that are memref casts and update function return types. More...
 

Typedef Documentation

◆ FuncCallerMap

using FuncCallerMap = DenseMap<func::FuncOp, DenseSet<Operation *> >

A mapping of FuncOps to their callers.

Definition at line 77 of file OneShotModuleBufferize.cpp.

Function Documentation

◆ equivalenceAnalysis()

static void equivalenceAnalysis ( func::FuncOp  funcOp,
OneShotAnalysisState state,
FuncAnalysisState funcState 
)
static

Gather equivalence info of CallOps.

Note: This only adds new equivalence info if the called function was already analyzed.

Definition at line 253 of file OneShotModuleBufferize.cpp.

◆ foldMemRefCasts()

static void foldMemRefCasts ( func::FuncOp  funcOp)
static

Fold return values that are memref casts and update function return types.

During FuncOp bufferization, the exact type of the returned memrefs (if any) is not known yet. Therefore, the bufferization uses memref types with the most generic layout map as function return types. After bufferizing the entire function body, a more concise memref type can potentially be used for the return type of the function.

Definition at line 350 of file OneShotModuleBufferize.cpp.

References mlir::get(), and getAssumedUniqueReturnOp().

Referenced by mlir::bufferization::bufferizeModuleOp().

◆ getAssumedUniqueReturnOp()

static func::ReturnOp getAssumedUniqueReturnOp ( func::FuncOp  funcOp)
static

Return the unique ReturnOp that terminates funcOp.

Return nullptr if there is no such unique ReturnOp.

Definition at line 90 of file OneShotModuleBufferize.cpp.

Referenced by foldMemRefCasts().

◆ getCalledFunction()

static func::FuncOp getCalledFunction ( func::CallOp  callOp)
static

Return the func::FuncOp called by callOp.

Definition at line 240 of file OneShotModuleBufferize.cpp.

References mlir::SymbolTable::lookupNearestSymbolFrom().

◆ getFuncOpsOrderedByCalls()

static LogicalResult getFuncOpsOrderedByCalls ( ModuleOp  moduleOp,
SmallVectorImpl< func::FuncOp > &  orderedFuncOps,
FuncCallerMap callerMap 
)
static

Store all functions of the moduleOp in orderedFuncOps, sorted by callee-caller order (i.e.

callees without callers first). Store the map of FuncOp to all its callers in callerMap. Return failure() if a cycle of calls is detected or if we are unable to retrieve the called FuncOp from any func::CallOp.

Definition at line 291 of file OneShotModuleBufferize.cpp.

References mlir::WalkResult::advance().

Referenced by mlir::bufferization::bufferizeModuleOp().

◆ getOrCreateFuncAnalysisState()

static FuncAnalysisState& getOrCreateFuncAnalysisState ( OneShotAnalysisState state)
static

Get or create FuncAnalysisState.

Definition at line 81 of file OneShotModuleBufferize.cpp.

◆ hasTensorSignature()

static bool hasTensorSignature ( func::FuncOp  funcOp)
static

Return "true" if the given function signature has tensor semantics.

Definition at line 279 of file OneShotModuleBufferize.cpp.

References isaTensor().

◆ removeBufferizationAttributes()

static void removeBufferizationAttributes ( BlockArgument  bbArg)
static

Remove bufferization attributes on FuncOp arguments.

Definition at line 231 of file OneShotModuleBufferize.cpp.

References mlir::BlockArgument::getArgNumber(), mlir::BlockArgument::getOwner(), and mlir::Block::getParentOp().