llvm-for-llvmta/include/llvm/IR/FixedPointBuilder.h

466 lines
19 KiB
C
Raw Permalink Normal View History

2022-04-25 10:02:23 +02:00
//===- llvm/FixedPointBuilder.h - Builder for fixed-point ops ---*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file defines the FixedPointBuilder class, which is used as a convenient
// way to lower fixed-point arithmetic operations to LLVM IR.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_IR_FIXEDPOINTBUILDER_H
#define LLVM_IR_FIXEDPOINTBUILDER_H
#include "llvm/ADT/APFixedPoint.h"
#include "llvm/IR/Constant.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/InstrTypes.h"
#include "llvm/IR/Instruction.h"
#include "llvm/IR/IntrinsicInst.h"
#include "llvm/IR/Intrinsics.h"
#include "llvm/IR/Type.h"
#include "llvm/IR/Value.h"
namespace llvm {
template <class IRBuilderTy> class FixedPointBuilder {
IRBuilderTy &B;
Value *Convert(Value *Src, const FixedPointSemantics &SrcSema,
const FixedPointSemantics &DstSema, bool DstIsInteger) {
unsigned SrcWidth = SrcSema.getWidth();
unsigned DstWidth = DstSema.getWidth();
unsigned SrcScale = SrcSema.getScale();
unsigned DstScale = DstSema.getScale();
bool SrcIsSigned = SrcSema.isSigned();
bool DstIsSigned = DstSema.isSigned();
Type *DstIntTy = B.getIntNTy(DstWidth);
Value *Result = Src;
unsigned ResultWidth = SrcWidth;
// Downscale.
if (DstScale < SrcScale) {
// When converting to integers, we round towards zero. For negative
// numbers, right shifting rounds towards negative infinity. In this case,
// we can just round up before shifting.
if (DstIsInteger && SrcIsSigned) {
Value *Zero = Constant::getNullValue(Result->getType());
Value *IsNegative = B.CreateICmpSLT(Result, Zero);
Value *LowBits = ConstantInt::get(
B.getContext(), APInt::getLowBitsSet(ResultWidth, SrcScale));
Value *Rounded = B.CreateAdd(Result, LowBits);
Result = B.CreateSelect(IsNegative, Rounded, Result);
}
Result = SrcIsSigned
? B.CreateAShr(Result, SrcScale - DstScale, "downscale")
: B.CreateLShr(Result, SrcScale - DstScale, "downscale");
}
if (!DstSema.isSaturated()) {
// Resize.
Result = B.CreateIntCast(Result, DstIntTy, SrcIsSigned, "resize");
// Upscale.
if (DstScale > SrcScale)
Result = B.CreateShl(Result, DstScale - SrcScale, "upscale");
} else {
// Adjust the number of fractional bits.
if (DstScale > SrcScale) {
// Compare to DstWidth to prevent resizing twice.
ResultWidth = std::max(SrcWidth + DstScale - SrcScale, DstWidth);
Type *UpscaledTy = B.getIntNTy(ResultWidth);
Result = B.CreateIntCast(Result, UpscaledTy, SrcIsSigned, "resize");
Result = B.CreateShl(Result, DstScale - SrcScale, "upscale");
}
// Handle saturation.
bool LessIntBits = DstSema.getIntegralBits() < SrcSema.getIntegralBits();
if (LessIntBits) {
Value *Max = ConstantInt::get(
B.getContext(),
APFixedPoint::getMax(DstSema).getValue().extOrTrunc(ResultWidth));
Value *TooHigh = SrcIsSigned ? B.CreateICmpSGT(Result, Max)
: B.CreateICmpUGT(Result, Max);
Result = B.CreateSelect(TooHigh, Max, Result, "satmax");
}
// Cannot overflow min to dest type if src is unsigned since all fixed
// point types can cover the unsigned min of 0.
if (SrcIsSigned && (LessIntBits || !DstIsSigned)) {
Value *Min = ConstantInt::get(
B.getContext(),
APFixedPoint::getMin(DstSema).getValue().extOrTrunc(ResultWidth));
Value *TooLow = B.CreateICmpSLT(Result, Min);
Result = B.CreateSelect(TooLow, Min, Result, "satmin");
}
// Resize the integer part to get the final destination size.
if (ResultWidth != DstWidth)
Result = B.CreateIntCast(Result, DstIntTy, SrcIsSigned, "resize");
}
return Result;
}
/// Get the common semantic for two semantics, with the added imposition that
/// saturated padded types retain the padding bit.
FixedPointSemantics
getCommonBinopSemantic(const FixedPointSemantics &LHSSema,
const FixedPointSemantics &RHSSema) {
auto C = LHSSema.getCommonSemantics(RHSSema);
bool BothPadded =
LHSSema.hasUnsignedPadding() && RHSSema.hasUnsignedPadding();
return FixedPointSemantics(
C.getWidth() + (unsigned)(BothPadded && C.isSaturated()), C.getScale(),
C.isSigned(), C.isSaturated(), BothPadded);
}
/// Given a floating point type and a fixed-point semantic, return a floating
/// point type which can accommodate the fixed-point semantic. This is either
/// \p Ty, or a floating point type with a larger exponent than Ty.
Type *getAccommodatingFloatType(Type *Ty, const FixedPointSemantics &Sema) {
const fltSemantics *FloatSema = &Ty->getFltSemantics();
while (!Sema.fitsInFloatSemantics(*FloatSema))
FloatSema = APFixedPoint::promoteFloatSemantics(FloatSema);
return Type::getFloatingPointTy(Ty->getContext(), *FloatSema);
}
public:
FixedPointBuilder(IRBuilderTy &Builder) : B(Builder) {}
/// Convert an integer value representing a fixed-point number from one
/// fixed-point semantic to another fixed-point semantic.
/// \p Src - The source value
/// \p SrcSema - The fixed-point semantic of the source value
/// \p DstSema - The resulting fixed-point semantic
Value *CreateFixedToFixed(Value *Src, const FixedPointSemantics &SrcSema,
const FixedPointSemantics &DstSema) {
return Convert(Src, SrcSema, DstSema, false);
}
/// Convert an integer value representing a fixed-point number to an integer
/// with the given bit width and signedness.
/// \p Src - The source value
/// \p SrcSema - The fixed-point semantic of the source value
/// \p DstWidth - The bit width of the result value
/// \p DstIsSigned - The signedness of the result value
Value *CreateFixedToInteger(Value *Src, const FixedPointSemantics &SrcSema,
unsigned DstWidth, bool DstIsSigned) {
return Convert(
Src, SrcSema,
FixedPointSemantics::GetIntegerSemantics(DstWidth, DstIsSigned), true);
}
/// Convert an integer value with the given signedness to an integer value
/// representing the given fixed-point semantic.
/// \p Src - The source value
/// \p SrcIsSigned - The signedness of the source value
/// \p DstSema - The resulting fixed-point semantic
Value *CreateIntegerToFixed(Value *Src, unsigned SrcIsSigned,
const FixedPointSemantics &DstSema) {
return Convert(Src,
FixedPointSemantics::GetIntegerSemantics(
Src->getType()->getScalarSizeInBits(), SrcIsSigned),
DstSema, false);
}
Value *CreateFixedToFloating(Value *Src, const FixedPointSemantics &SrcSema,
Type *DstTy) {
Value *Result;
Type *OpTy = getAccommodatingFloatType(DstTy, SrcSema);
// Convert the raw fixed-point value directly to floating point. If the
// value is too large to fit, it will be rounded, not truncated.
Result = SrcSema.isSigned() ? B.CreateSIToFP(Src, OpTy)
: B.CreateUIToFP(Src, OpTy);
// Rescale the integral-in-floating point by the scaling factor. This is
// lossless, except for overflow to infinity which is unlikely.
Result = B.CreateFMul(Result,
ConstantFP::get(OpTy, std::pow(2, -(int)SrcSema.getScale())));
if (OpTy != DstTy)
Result = B.CreateFPTrunc(Result, DstTy);
return Result;
}
Value *CreateFloatingToFixed(Value *Src, const FixedPointSemantics &DstSema) {
bool UseSigned = DstSema.isSigned() || DstSema.hasUnsignedPadding();
Value *Result = Src;
Type *OpTy = getAccommodatingFloatType(Src->getType(), DstSema);
if (OpTy != Src->getType())
Result = B.CreateFPExt(Result, OpTy);
// Rescale the floating point value so that its significant bits (for the
// purposes of the conversion) are in the integral range.
Result = B.CreateFMul(Result,
ConstantFP::get(OpTy, std::pow(2, DstSema.getScale())));
Type *ResultTy = B.getIntNTy(DstSema.getWidth());
if (DstSema.isSaturated()) {
Intrinsic::ID IID =
UseSigned ? Intrinsic::fptosi_sat : Intrinsic::fptoui_sat;
Result = B.CreateIntrinsic(IID, {ResultTy, OpTy}, {Result});
} else {
Result = UseSigned ? B.CreateFPToSI(Result, ResultTy)
: B.CreateFPToUI(Result, ResultTy);
}
// When saturating unsigned-with-padding using signed operations, we may
// get negative values. Emit an extra clamp to zero.
if (DstSema.isSaturated() && DstSema.hasUnsignedPadding()) {
Constant *Zero = Constant::getNullValue(Result->getType());
Result =
B.CreateSelect(B.CreateICmpSLT(Result, Zero), Zero, Result, "satmin");
}
return Result;
}
/// Add two fixed-point values and return the result in their common semantic.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateAdd(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
bool UseSigned = CommonSema.isSigned() || CommonSema.hasUnsignedPadding();
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
Value *Result;
if (CommonSema.isSaturated()) {
Intrinsic::ID IID = UseSigned ? Intrinsic::sadd_sat : Intrinsic::uadd_sat;
Result = B.CreateBinaryIntrinsic(IID, WideLHS, WideRHS);
} else {
Result = B.CreateAdd(WideLHS, WideRHS);
}
return CreateFixedToFixed(Result, CommonSema,
LHSSema.getCommonSemantics(RHSSema));
}
/// Subtract two fixed-point values and return the result in their common
/// semantic.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateSub(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
bool UseSigned = CommonSema.isSigned() || CommonSema.hasUnsignedPadding();
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
Value *Result;
if (CommonSema.isSaturated()) {
Intrinsic::ID IID = UseSigned ? Intrinsic::ssub_sat : Intrinsic::usub_sat;
Result = B.CreateBinaryIntrinsic(IID, WideLHS, WideRHS);
} else {
Result = B.CreateSub(WideLHS, WideRHS);
}
// Subtraction can end up below 0 for padded unsigned operations, so emit
// an extra clamp in that case.
if (CommonSema.isSaturated() && CommonSema.hasUnsignedPadding()) {
Constant *Zero = Constant::getNullValue(Result->getType());
Result =
B.CreateSelect(B.CreateICmpSLT(Result, Zero), Zero, Result, "satmin");
}
return CreateFixedToFixed(Result, CommonSema,
LHSSema.getCommonSemantics(RHSSema));
}
/// Multiply two fixed-point values and return the result in their common
/// semantic.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateMul(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
bool UseSigned = CommonSema.isSigned() || CommonSema.hasUnsignedPadding();
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
Intrinsic::ID IID;
if (CommonSema.isSaturated()) {
IID = UseSigned ? Intrinsic::smul_fix_sat : Intrinsic::umul_fix_sat;
} else {
IID = UseSigned ? Intrinsic::smul_fix : Intrinsic::umul_fix;
}
Value *Result = B.CreateIntrinsic(
IID, {WideLHS->getType()},
{WideLHS, WideRHS, B.getInt32(CommonSema.getScale())});
return CreateFixedToFixed(Result, CommonSema,
LHSSema.getCommonSemantics(RHSSema));
}
/// Divide two fixed-point values and return the result in their common
/// semantic.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateDiv(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
bool UseSigned = CommonSema.isSigned() || CommonSema.hasUnsignedPadding();
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
Intrinsic::ID IID;
if (CommonSema.isSaturated()) {
IID = UseSigned ? Intrinsic::sdiv_fix_sat : Intrinsic::udiv_fix_sat;
} else {
IID = UseSigned ? Intrinsic::sdiv_fix : Intrinsic::udiv_fix;
}
Value *Result = B.CreateIntrinsic(
IID, {WideLHS->getType()},
{WideLHS, WideRHS, B.getInt32(CommonSema.getScale())});
return CreateFixedToFixed(Result, CommonSema,
LHSSema.getCommonSemantics(RHSSema));
}
/// Left shift a fixed-point value by an unsigned integer value. The integer
/// value can be any bit width.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
Value *CreateShl(Value *LHS, const FixedPointSemantics &LHSSema, Value *RHS) {
bool UseSigned = LHSSema.isSigned() || LHSSema.hasUnsignedPadding();
RHS = B.CreateIntCast(RHS, LHS->getType(), /*IsSigned=*/false);
Value *Result;
if (LHSSema.isSaturated()) {
Intrinsic::ID IID = UseSigned ? Intrinsic::sshl_sat : Intrinsic::ushl_sat;
Result = B.CreateBinaryIntrinsic(IID, LHS, RHS);
} else {
Result = B.CreateShl(LHS, RHS);
}
return Result;
}
/// Right shift a fixed-point value by an unsigned integer value. The integer
/// value can be any bit width.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
Value *CreateShr(Value *LHS, const FixedPointSemantics &LHSSema, Value *RHS) {
RHS = B.CreateIntCast(RHS, LHS->getType(), false);
return LHSSema.isSigned() ? B.CreateAShr(LHS, RHS) : B.CreateLShr(LHS, RHS);
}
/// Compare two fixed-point values for equality.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateEQ(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return B.CreateICmpEQ(WideLHS, WideRHS);
}
/// Compare two fixed-point values for inequality.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateNE(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return B.CreateICmpNE(WideLHS, WideRHS);
}
/// Compare two fixed-point values as LHS < RHS.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateLT(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return CommonSema.isSigned() ? B.CreateICmpSLT(WideLHS, WideRHS)
: B.CreateICmpULT(WideLHS, WideRHS);
}
/// Compare two fixed-point values as LHS <= RHS.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateLE(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return CommonSema.isSigned() ? B.CreateICmpSLE(WideLHS, WideRHS)
: B.CreateICmpULE(WideLHS, WideRHS);
}
/// Compare two fixed-point values as LHS > RHS.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateGT(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return CommonSema.isSigned() ? B.CreateICmpSGT(WideLHS, WideRHS)
: B.CreateICmpUGT(WideLHS, WideRHS);
}
/// Compare two fixed-point values as LHS >= RHS.
/// \p LHS - The left hand side
/// \p LHSSema - The semantic of the left hand side
/// \p RHS - The right hand side
/// \p RHSSema - The semantic of the right hand side
Value *CreateGE(Value *LHS, const FixedPointSemantics &LHSSema,
Value *RHS, const FixedPointSemantics &RHSSema) {
auto CommonSema = getCommonBinopSemantic(LHSSema, RHSSema);
Value *WideLHS = CreateFixedToFixed(LHS, LHSSema, CommonSema);
Value *WideRHS = CreateFixedToFixed(RHS, RHSSema, CommonSema);
return CommonSema.isSigned() ? B.CreateICmpSGE(WideLHS, WideRHS)
: B.CreateICmpUGE(WideLHS, WideRHS);
}
};
} // end namespace llvm
#endif // LLVM_IR_FIXEDPOINTBUILDER_H