MathRsqrtApproximation.cpp File Reference

#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Math/IR/Math.h"
#include "mlir/Dialect/Utils/IndexingUtils.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/X86/Transforms.h"
#include "mlir/Dialect/X86/X86Dialect.h"
#include "mlir/IR/ImplicitLocOpBuilder.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/IR/TypeUtilities.h"

Go to the source code of this file.

Functions
static Value	handleMultidimensionalVectors (ImplicitLocOpBuilder &builder, ValueRange operands, int64_t vectorWidth, llvm::function_ref< Value(ValueRange)> compute)

Function Documentation

handleMultidimensionalVectors()

Value handleMultidimensionalVectors ( ImplicitLocOpBuilder & builder, ValueRange operands, int64_t vectorWidth, llvm::function_ref< Value(ValueRange)> compute )

static

Definition at line 39 of file MathRsqrtApproximation.cpp.

References mlir::computeMaxLinearIndex(), mlir::computeStrides(), mlir::delinearize(), mlir::getType(), mlir::Builder::getZeroAttr(), and result.

Referenced by mlir::LLVM::detail::vectorOneToOneRewrite().