20 #define GEN_PASS_DEF_LOOPINVARIANTCODEMOTION
21 #define GEN_PASS_DEF_LOOPINVARIANTSUBSETHOISTING
22 #include "mlir/Transforms/Passes.h.inc"
29 struct LoopInvariantCodeMotion
30 :
public impl::LoopInvariantCodeMotionBase<LoopInvariantCodeMotion> {
31 void runOnOperation()
override;
34 struct LoopInvariantSubsetHoisting
35 :
public impl::LoopInvariantSubsetHoistingBase<
36 LoopInvariantSubsetHoisting> {
37 void runOnOperation()
override;
41 void LoopInvariantCodeMotion::runOnOperation() {
49 void LoopInvariantSubsetHoisting::runOnOperation() {
54 getOperation()->walk([&](LoopLikeOpInterface loopLike) {
60 return std::make_unique<LoopInvariantCodeMotion>();
64 return std::make_unique<LoopInvariantSubsetHoisting>();
static MLIRContext * getContext(OpFoldResult val)
This class coordinates rewriting a piece of IR outside of a pattern rewrite, providing a way to keep ...
Include the generated interface declarations.
LoopLikeOpInterface hoistLoopInvariantSubsets(RewriterBase &rewriter, LoopLikeOpInterface loopLike)
Hoist loop-invariant tensor subsets (subset extraction and subset insertion ops) from loop-like ops.
std::unique_ptr< Pass > createLoopInvariantCodeMotionPass()
Creates a loop invariant code motion pass that hoists loop invariant instructions out of the loop.
std::unique_ptr< Pass > createLoopInvariantSubsetHoistingPass()
Creates a pass that hoists loop-invariant subset ops.
size_t moveLoopInvariantCode(ArrayRef< Region * > regions, function_ref< bool(Value, Region *)> isDefinedOutsideRegion, function_ref< bool(Operation *, Region *)> shouldMoveOutOfRegion, function_ref< void(Operation *, Region *)> moveOutOfRegion)
Given a list of regions, perform loop-invariant code motion.