EmulateWideInt.cpp File Reference

#include "mlir/Dialect/Arith/Transforms/Passes.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Transforms/WideIntEmulationConverter.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/Func/Transforms/FuncConversions.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/TypeUtilities.h"
#include "mlir/Transforms/DialectConversion.h"
#include "llvm/ADT/APInt.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/Support/MathExtras.h"
#include <cassert>
#include "mlir/Dialect/Arith/Transforms/Passes.h.inc"

Go to the source code of this file.

Namespaces
	mlir
	Include the generated interface declarations.
	mlir::arith

Macros
#define	GEN_PASS_DEF_ARITHEMULATEWIDEINT

Functions
static std::pair< APInt, APInt >	getHalves (const APInt &value, unsigned newBitWidth)
	Returns N bottom and N top bits from value, where N = newBitWidth. More...
static Type	reduceInnermostDim (VectorType type)
	Returns the type with the last (innermost) dimension reduced to x1. More...
static Value	extractLastDimSlice (ConversionPatternRewriter &rewriter, Location loc, Value input, int64_t lastOffset)
	Extracts the input vector slice with elements at the last dimension offset by lastOffset. More...
static std::pair< Value, Value >	extractLastDimHalves (ConversionPatternRewriter &rewriter, Location loc, Value input)
	Extracts two vector slices from the input whose type is vector<...x2T>, with the first element at offset 0 and the second element at offset 1. More...
static Value	dropTrailingX1Dim (ConversionPatternRewriter &rewriter, Location loc, Value input)
static Value	appendX1Dim (ConversionPatternRewriter &rewriter, Location loc, Value input)
	Performs a vector shape cast to append an x1 dimension. More...
static Value	insertLastDimSlice (ConversionPatternRewriter &rewriter, Location loc, Value source, Value dest, int64_t lastOffset)
	Inserts the source vector slice into the dest vector at offset lastOffset in the last dimension. More...
static Value	constructResultVector (ConversionPatternRewriter &rewriter, Location loc, VectorType resultType, ValueRange resultComponents)
	Constructs a new vector of type resultType by creating a series of insertions of resultComponents, each at the next offset of the last vector dimension. More...

Macro Definition Documentation

GEN_PASS_DEF_ARITHEMULATEWIDEINT

#define GEN_PASS_DEF_ARITHEMULATEWIDEINT

Definition at line 26 of file EmulateWideInt.cpp.

Function Documentation

appendX1Dim()

static Value appendX1Dim ( ConversionPatternRewriter &  rewriter, Location  loc, Value  input  )

static

Performs a vector shape cast to append an x1 dimension.

If the input is a scalar, this is a noop.

Definition at line 114 of file EmulateWideInt.cpp.

References mlir::OpBuilder::create(), mlir::get(), and mlir::Value::getType().

constructResultVector()

static Value constructResultVector ( ConversionPatternRewriter &  rewriter, Location  loc, VectorType  resultType, ValueRange  resultComponents  )

static

Constructs a new vector of type resultType by creating a series of insertions of resultComponents, each at the next offset of the last vector dimension.

When all resultComponents are scalars, the result type is vector<NxT>; when resultComponents are vector<...x1xT>s, the result type is vector<...xNxT>, where N is the number of resultComponents.

Definition at line 153 of file EmulateWideInt.cpp.

References mlir::createScalarOrSplatConstant(), mlir::detail::enumerate(), and insertLastDimSlice().

dropTrailingX1Dim()

static Value dropTrailingX1Dim ( ConversionPatternRewriter &  rewriter, Location  loc, Value  input  )

static

Definition at line 97 of file EmulateWideInt.cpp.

References mlir::OpBuilder::create(), mlir::get(), and mlir::Value::getType().

extractLastDimHalves()

static std::pair<Value, Value> extractLastDimHalves ( ConversionPatternRewriter &  rewriter, Location  loc, Value  input  )

static

Extracts two vector slices from the input whose type is vector<...x2T>, with the first element at offset 0 and the second element at offset 1.

Definition at line 89 of file EmulateWideInt.cpp.

References extractLastDimSlice().

extractLastDimSlice()

static Value extractLastDimSlice ( ConversionPatternRewriter &  rewriter, Location  loc, Value  input, int64_t  lastOffset  )

static

Extracts the input vector slice with elements at the last dimension offset by lastOffset.

Returns a value of vector type with the last dimension reduced to x1 or fully scalarized, e.g.:

• vector<3x2xi16> --> vector<3x1xi16>
• vector<2xi16> --> i16

Definition at line 66 of file EmulateWideInt.cpp.

References mlir::OpBuilder::create(), and mlir::Value::getType().

Referenced by extractLastDimHalves().

getHalves()

static std::pair<APInt, APInt> getHalves ( const APInt &  value, unsigned  newBitWidth  )

static

Returns N bottom and N top bits from value, where N = newBitWidth.

Treats value as a 2*N bits-wide integer. The bottom bits are returned in the first pair element, while the top bits in the second one.

Definition at line 40 of file EmulateWideInt.cpp.

insertLastDimSlice()

static Value insertLastDimSlice ( ConversionPatternRewriter &  rewriter, Location  loc, Value  source, Value  dest, int64_t  lastOffset  )

static

Inserts the source vector slice into the dest vector at offset lastOffset in the last dimension.

source can be a scalar when dest is a 1D vector.

Definition at line 130 of file EmulateWideInt.cpp.

References mlir::OpBuilder::create(), and mlir::Value::getType().

Referenced by constructResultVector().

reduceInnermostDim()

static Type reduceInnermostDim ( VectorType  type )

static

Returns the type with the last (innermost) dimension reduced to x1.

Scalarizes 1D vector inputs to match how we extract/insert vector values, e.g.:

• vector<3x2xi16> --> vector<3x1xi16>
• vector<2xi16> --> i16

Definition at line 52 of file EmulateWideInt.cpp.

References mlir::get().