OpenACCUtilsTiling.cpp File Reference

#include "mlir/Dialect/OpenACC/OpenACCUtilsTiling.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/OpenACC/OpenACC.h"
#include "mlir/Dialect/Utils/StaticValueUtils.h"
#include "mlir/Transforms/RegionUtils.h"
#include "llvm/ADT/STLExtras.h"

Go to the source code of this file.

Functions
static mlir::Value	resolveAndCastTileSize (mlir::Value tileSize, int32_t defaultTileSize, mlir::Type targetType, mlir::RewriterBase &rewriter, mlir::Location loc)
static void	removeWorkerVectorFromLoop (mlir::acc::LoopOp loop)
static mlir::acc::LoopOp	createACCLoopFromOriginal (mlir::acc::LoopOp origLoop, mlir::RewriterBase &rewriter, mlir::ValueRange lb, mlir::ValueRange ub, mlir::ValueRange step, mlir::DenseBoolArrayAttr inclusiveUBAttr, mlir::acc::CombinedConstructsTypeAttr combinedAttr, mlir::Location loc)
static void	moveOpsAndReplaceIVs (mlir::acc::LoopOp sourceLoop, mlir::acc::LoopOp targetLoop, llvm::ArrayRef< mlir::Value > newIVs, llvm::ArrayRef< mlir::Value > origIVs, size_t nOps, mlir::RewriterBase &rewriter)
static mlir::acc::LoopOp	createElementGroupLoop (mlir::acc::LoopOp tileLoop, mlir::RewriterBase &rewriter, mlir::ValueRange lbs, mlir::ValueRange ubs, mlir::ValueRange steps, mlir::DenseBoolArrayAttr inclusiveUBAttr, llvm::ArrayRef< mlir::Type > ivTypes, mlir::Location loc)

Function Documentation

createACCLoopFromOriginal()

mlir::acc::LoopOp createACCLoopFromOriginal ( mlir::acc::LoopOp origLoop, mlir::RewriterBase & rewriter, mlir::ValueRange lb, mlir::ValueRange ub, mlir::ValueRange step, mlir::DenseBoolArrayAttr inclusiveUBAttr, mlir::acc::CombinedConstructsTypeAttr combinedAttr, mlir::Location loc )

static

Definition at line 52 of file OpenACCUtilsTiling.cpp.

Referenced by createElementGroupLoop().

createElementGroupLoop()

mlir::acc::LoopOp createElementGroupLoop ( mlir::acc::LoopOp tileLoop, mlir::RewriterBase & rewriter, mlir::ValueRange lbs, mlir::ValueRange ubs, mlir::ValueRange steps, mlir::DenseBoolArrayAttr inclusiveUBAttr, llvm::ArrayRef< mlir::Type > ivTypes, mlir::Location loc )

static

Definition at line 114 of file OpenACCUtilsTiling.cpp.

References createACCLoopFromOriginal(), mlir::OpBuilder::createBlock(), mlir::RewriterBase::finalizeOpModification(), mlir::OpBuilder::setInsertionPointToEnd(), and mlir::RewriterBase::startOpModification().

Referenced by mlir::acc::tileACCLoops().

moveOpsAndReplaceIVs()

void moveOpsAndReplaceIVs ( mlir::acc::LoopOp sourceLoop, mlir::acc::LoopOp targetLoop, llvm::ArrayRef< mlir::Value > newIVs, llvm::ArrayRef< mlir::Value > origIVs, size_t nOps, mlir::RewriterBase & rewriter )

static

Definition at line 77 of file OpenACCUtilsTiling.cpp.

References mlir::RewriterBase::finalizeOpModification(), and mlir::replaceAllUsesInRegionWith().

Referenced by mlir::acc::tileACCLoops().

removeWorkerVectorFromLoop()

void removeWorkerVectorFromLoop ( mlir::acc::LoopOp loop )

static

Definition at line 41 of file OpenACCUtilsTiling.cpp.

Referenced by mlir::acc::tileACCLoops().

resolveAndCastTileSize()

mlir::Value resolveAndCastTileSize ( mlir::Value tileSize, int32_t defaultTileSize, mlir::Type targetType, mlir::RewriterBase & rewriter, mlir::Location loc )

static

Definition at line 24 of file OpenACCUtilsTiling.cpp.

References mlir::getConstantIntValue(), mlir::Builder::getIntegerAttr(), and mlir::getValueOrCreateCastToIndexLike().

Referenced by mlir::acc::tileACCLoops().