xref: /freebsd/contrib/llvm-project/llvm/lib/Target/BPF/BPFTargetTransformInfo.h (revision 924226fba12cc9a228c73b956e1b7fa24c60b055)
1 //===------ BPFTargetTransformInfo.h - BPF specific TTI ---------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file uses the target's specific information to
10 // provide more precise answers to certain TTI queries, while letting the
11 // target independent and default TTI implementations handle the rest.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef LLVM_LIB_TARGET_BPF_BPFTARGETTRANSFORMINFO_H
16 #define LLVM_LIB_TARGET_BPF_BPFTARGETTRANSFORMINFO_H
17 
18 #include "BPFTargetMachine.h"
19 #include "llvm/Analysis/TargetTransformInfo.h"
20 #include "llvm/CodeGen/BasicTTIImpl.h"
21 #include "llvm/Transforms/Utils/ScalarEvolutionExpander.h"
22 
23 namespace llvm {
24 class BPFTTIImpl : public BasicTTIImplBase<BPFTTIImpl> {
25   typedef BasicTTIImplBase<BPFTTIImpl> BaseT;
26   typedef TargetTransformInfo TTI;
27   friend BaseT;
28 
29   const BPFSubtarget *ST;
30   const BPFTargetLowering *TLI;
31 
32   const BPFSubtarget *getST() const { return ST; }
33   const BPFTargetLowering *getTLI() const { return TLI; }
34 
35 public:
36   explicit BPFTTIImpl(const BPFTargetMachine *TM, const Function &F)
37       : BaseT(TM, F.getParent()->getDataLayout()), ST(TM->getSubtargetImpl(F)),
38         TLI(ST->getTargetLowering()) {}
39 
40   int getIntImmCost(const APInt &Imm, Type *Ty, TTI::TargetCostKind CostKind) {
41     if (Imm.getBitWidth() <= 64 && isInt<32>(Imm.getSExtValue()))
42       return TTI::TCC_Free;
43 
44     return TTI::TCC_Basic;
45   }
46 
47   InstructionCost getCmpSelInstrCost(unsigned Opcode, Type *ValTy, Type *CondTy,
48                                      CmpInst::Predicate VecPred,
49                                      TTI::TargetCostKind CostKind,
50                                      const llvm::Instruction *I = nullptr) {
51     if (Opcode == Instruction::Select)
52       return SCEVCheapExpansionBudget.getValue();
53 
54     return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy, VecPred, CostKind,
55                                      I);
56   }
57 
58   InstructionCost getArithmeticInstrCost(
59       unsigned Opcode, Type *Ty, TTI::TargetCostKind CostKind,
60       TTI::OperandValueKind Opd1Info = TTI::OK_AnyValue,
61       TTI::OperandValueKind Opd2Info = TTI::OK_AnyValue,
62       TTI::OperandValueProperties Opd1PropInfo = TTI::OP_None,
63       TTI::OperandValueProperties Opd2PropInfo = TTI::OP_None,
64     ArrayRef<const Value *> Args = ArrayRef<const Value *>(),
65     const Instruction *CxtI = nullptr) {
66       int ISD = TLI->InstructionOpcodeToISD(Opcode);
67       if (ISD == ISD::ADD && CostKind == TTI::TCK_RecipThroughput)
68         return SCEVCheapExpansionBudget.getValue() + 1;
69 
70       return BaseT::getArithmeticInstrCost(Opcode, Ty, CostKind, Opd1Info,
71                                            Opd2Info, Opd1PropInfo,
72                                            Opd2PropInfo);
73   }
74 };
75 
76 } // end namespace llvm
77 
78 #endif // LLVM_LIB_TARGET_BPF_BPFTARGETTRANSFORMINFO_H
79