VectorToGPUUtils.cpp File Reference

#include "mlir/Conversion/VectorToGPUCommon/VectorToGPUUtils.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/PatternMatch.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/TypeSwitch.h"
#include "llvm/Support/DebugLog.h"
#include <cstddef>
#include <tuple>
#include <utility>

Go to the source code of this file.

Macros
#define	DEBUG_TYPE   "vector-to-gpu-common"

Functions
template bool	mlir::integerExtendSupportsMMAMatrixType< arith::ExtSIOp > (arith::ExtSIOp)
template bool	mlir::integerExtendSupportsMMAMatrixType< arith::ExtUIOp > (arith::ExtUIOp)
static scf::ForOp	replaceForOpWithNewSignature (RewriterBase &rewriter, scf::ForOp loop, ValueRange newInitArgs)

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "vector-to-gpu-common"

Definition at line 25 of file VectorToGPUUtils.cpp.

Function Documentation

mlir::integerExtendSupportsMMAMatrixType< arith::ExtSIOp >()

template bool mlir::integerExtendSupportsMMAMatrixType< arith::ExtSIOp >

(

arith::ExtSIOp

)

mlir::integerExtendSupportsMMAMatrixType< arith::ExtUIOp >()

template bool mlir::integerExtendSupportsMMAMatrixType< arith::ExtUIOp >

(

arith::ExtUIOp

)

replaceForOpWithNewSignature()

scf::ForOp replaceForOpWithNewSignature ( RewriterBase & rewriter, scf::ForOp loop, ValueRange newInitArgs )

static

Definition at line 82 of file VectorToGPUUtils.cpp.

References mlir::RewriterBase::eraseBlock(), mlir::RewriterBase::eraseOp(), mlir::RewriterBase::replaceAllUsesWith(), and mlir::OpBuilder::setInsertionPoint().

Referenced by mlir::convertForOpForMMA().