256 lines
9.5 KiB
C
256 lines
9.5 KiB
C
|
//===- InstSimplifyFolder.h - InstSimplify folding helper --------*- 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 InstSimplifyFolder class, a helper for IRBuilder.
|
||
|
|
// It provides IRBuilder with a set of methods for folding operations to
|
||
|
|
// existing values using InstructionSimplify. At the moment, only a subset of
|
||
|
|
// the implementation uses InstructionSimplify. The rest of the implementation
|
||
|
|
// only folds constants.
|
||
|
|
//
|
||
|
|
// The folder also applies target-specific constant folding.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#ifndef LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H
|
||
|
|
#define LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H
|
||
|
|
|
||
|
|
#include "llvm/ADT/ArrayRef.h"
|
||
|
|
#include "llvm/Analysis/InstructionSimplify.h"
|
||
|
|
#include "llvm/Analysis/TargetFolder.h"
|
||
|
|
#include "llvm/IR/Constants.h"
|
||
|
|
#include "llvm/IR/IRBuilderFolder.h"
|
||
|
|
#include "llvm/IR/InstrTypes.h"
|
||
|
|
#include "llvm/IR/Instruction.h"
|
||
|
|
|
||
|
|
namespace llvm {
|
||
|
|
|
||
|
|
/// InstSimplifyFolder - Use InstructionSimplify to fold operations to existing
|
||
|
|
/// values. Also applies target-specific constant folding when not using
|
||
|
|
/// InstructionSimplify.
|
||
|
|
class InstSimplifyFolder final : public IRBuilderFolder {
|
||
|
|
TargetFolder ConstFolder;
|
||
|
|
SimplifyQuery SQ;
|
||
|
|
|
||
|
|
virtual void anchor();
|
||
|
|
|
||
|
|
public:
|
||
|
|
InstSimplifyFolder(const DataLayout &DL) : ConstFolder(DL), SQ(DL) {}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Value-based folders.
|
||
|
|
//
|
||
|
|
// Return an existing value or a constant if the operation can be simplified.
|
||
|
|
// Otherwise return nullptr.
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
Value *FoldAdd(Value *LHS, Value *RHS, bool HasNUW = false,
|
||
|
|
bool HasNSW = false) const override {
|
||
|
|
return SimplifyAddInst(LHS, RHS, HasNUW, HasNSW, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *FoldAnd(Value *LHS, Value *RHS) const override {
|
||
|
|
return SimplifyAndInst(LHS, RHS, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *FoldOr(Value *LHS, Value *RHS) const override {
|
||
|
|
return SimplifyOrInst(LHS, RHS, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *FoldICmp(CmpInst::Predicate P, Value *LHS, Value *RHS) const override {
|
||
|
|
return SimplifyICmpInst(P, LHS, RHS, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *FoldGEP(Type *Ty, Value *Ptr, ArrayRef<Value *> IdxList,
|
||
|
|
bool IsInBounds = false) const override {
|
||
|
|
return SimplifyGEPInst(Ty, Ptr, IdxList, IsInBounds, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *FoldSelect(Value *C, Value *True, Value *False) const override {
|
||
|
|
return SimplifySelectInst(C, True, False, SQ);
|
||
|
|
}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Binary Operators
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
Value *CreateFAdd(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFAdd(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateSub(Constant *LHS, Constant *RHS, bool HasNUW = false,
|
||
|
|
bool HasNSW = false) const override {
|
||
|
|
return ConstFolder.CreateSub(LHS, RHS, HasNUW, HasNSW);
|
||
|
|
}
|
||
|
|
Value *CreateFSub(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFSub(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateMul(Constant *LHS, Constant *RHS, bool HasNUW = false,
|
||
|
|
bool HasNSW = false) const override {
|
||
|
|
return ConstFolder.CreateMul(LHS, RHS, HasNUW, HasNSW);
|
||
|
|
}
|
||
|
|
Value *CreateFMul(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFMul(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateUDiv(Constant *LHS, Constant *RHS,
|
||
|
|
bool isExact = false) const override {
|
||
|
|
return ConstFolder.CreateUDiv(LHS, RHS, isExact);
|
||
|
|
}
|
||
|
|
Value *CreateSDiv(Constant *LHS, Constant *RHS,
|
||
|
|
bool isExact = false) const override {
|
||
|
|
return ConstFolder.CreateSDiv(LHS, RHS, isExact);
|
||
|
|
}
|
||
|
|
Value *CreateFDiv(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFDiv(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateURem(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateURem(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateSRem(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateSRem(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateFRem(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFRem(LHS, RHS);
|
||
|
|
}
|
||
|
|
Value *CreateShl(Constant *LHS, Constant *RHS, bool HasNUW = false,
|
||
|
|
bool HasNSW = false) const override {
|
||
|
|
return ConstFolder.CreateShl(LHS, RHS, HasNUW, HasNSW);
|
||
|
|
}
|
||
|
|
Value *CreateLShr(Constant *LHS, Constant *RHS,
|
||
|
|
bool isExact = false) const override {
|
||
|
|
return ConstFolder.CreateLShr(LHS, RHS, isExact);
|
||
|
|
}
|
||
|
|
Value *CreateAShr(Constant *LHS, Constant *RHS,
|
||
|
|
bool isExact = false) const override {
|
||
|
|
return ConstFolder.CreateAShr(LHS, RHS, isExact);
|
||
|
|
}
|
||
|
|
Value *CreateXor(Constant *LHS, Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateXor(LHS, RHS);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateBinOp(Instruction::BinaryOps Opc, Constant *LHS,
|
||
|
|
Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateBinOp(Opc, LHS, RHS);
|
||
|
|
}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Unary Operators
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
Value *CreateNeg(Constant *C, bool HasNUW = false,
|
||
|
|
bool HasNSW = false) const override {
|
||
|
|
return ConstFolder.CreateNeg(C, HasNUW, HasNSW);
|
||
|
|
}
|
||
|
|
Value *CreateFNeg(Constant *C) const override {
|
||
|
|
return ConstFolder.CreateFNeg(C);
|
||
|
|
}
|
||
|
|
Value *CreateNot(Constant *C) const override {
|
||
|
|
return ConstFolder.CreateNot(C);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateUnOp(Instruction::UnaryOps Opc, Constant *C) const override {
|
||
|
|
return ConstFolder.CreateUnOp(Opc, C);
|
||
|
|
}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Cast/Conversion Operators
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
Value *CreateCast(Instruction::CastOps Op, Constant *C,
|
||
|
|
Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateCast(Op, C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateIntCast(Constant *C, Type *DestTy,
|
||
|
|
bool isSigned) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateIntCast(C, DestTy, isSigned);
|
||
|
|
}
|
||
|
|
Value *CreatePointerCast(Constant *C, Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreatePointerCast(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateFPCast(Constant *C, Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateFPCast(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateBitCast(Constant *C, Type *DestTy) const override {
|
||
|
|
return ConstFolder.CreateBitCast(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateIntToPtr(Constant *C, Type *DestTy) const override {
|
||
|
|
return ConstFolder.CreateIntToPtr(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreatePtrToInt(Constant *C, Type *DestTy) const override {
|
||
|
|
return ConstFolder.CreatePtrToInt(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateZExtOrBitCast(Constant *C, Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateZExtOrBitCast(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateSExtOrBitCast(Constant *C, Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateSExtOrBitCast(C, DestTy);
|
||
|
|
}
|
||
|
|
Value *CreateTruncOrBitCast(Constant *C, Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreateTruncOrBitCast(C, DestTy);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreatePointerBitCastOrAddrSpaceCast(Constant *C,
|
||
|
|
Type *DestTy) const override {
|
||
|
|
if (C->getType() == DestTy)
|
||
|
|
return C; // avoid calling Fold
|
||
|
|
return ConstFolder.CreatePointerBitCastOrAddrSpaceCast(C, DestTy);
|
||
|
|
}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Compare Instructions
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
Value *CreateFCmp(CmpInst::Predicate P, Constant *LHS,
|
||
|
|
Constant *RHS) const override {
|
||
|
|
return ConstFolder.CreateFCmp(P, LHS, RHS);
|
||
|
|
}
|
||
|
|
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
// Other Instructions
|
||
|
|
//===--------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
Value *CreateExtractElement(Constant *Vec, Constant *Idx) const override {
|
||
|
|
return ConstFolder.CreateExtractElement(Vec, Idx);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateInsertElement(Constant *Vec, Constant *NewElt,
|
||
|
|
Constant *Idx) const override {
|
||
|
|
return ConstFolder.CreateInsertElement(Vec, NewElt, Idx);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateShuffleVector(Constant *V1, Constant *V2,
|
||
|
|
ArrayRef<int> Mask) const override {
|
||
|
|
return ConstFolder.CreateShuffleVector(V1, V2, Mask);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateExtractValue(Constant *Agg,
|
||
|
|
ArrayRef<unsigned> IdxList) const override {
|
||
|
|
return ConstFolder.CreateExtractValue(Agg, IdxList);
|
||
|
|
}
|
||
|
|
|
||
|
|
Value *CreateInsertValue(Constant *Agg, Constant *Val,
|
||
|
|
ArrayRef<unsigned> IdxList) const override {
|
||
|
|
return ConstFolder.CreateInsertValue(Agg, Val, IdxList);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // end namespace llvm
|
||
|
|
|
||
|
|
#endif // LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H
|