[CallSite removal][TargetTransformInfoImpl] Replace CallSite with Cal…

…lBase. NFC
llvm · Apr 22, 2020 · daadb48 · daadb48
1 parent c08af2c
commit daadb48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h b/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
@@ -17,7 +17,6 @@
 #include "llvm/Analysis/ScalarEvolutionExpressions.h"
 #include "llvm/Analysis/TargetTransformInfo.h"
 #include "llvm/Analysis/VectorUtils.h"
-#include "llvm/IR/CallSite.h"
 #include "llvm/IR/DataLayout.h"
 #include "llvm/IR/Function.h"
 #include "llvm/IR/GetElementPtrTypeIterator.h"
@@ -796,8 +795,8 @@ class TargetTransformInfoImplCRTPBase : public TargetTransformInfoImplBase {
   unsigned getUserCost(const User *U, ArrayRef<const Value *> Operands) {
     auto *TargetTTI = static_cast<T *>(this);
 
-    if (auto CS = ImmutableCallSite(U)) {
-      const Function *F = CS.getCalledFunction();
+    if (const auto *CB = dyn_cast<CallBase>(U)) {
+      const Function *F = CB->getCalledFunction();
       if (F) {
         FunctionType *FTy = F->getFunctionType();
         if (Intrinsic::ID IID = F->getIntrinsicID()) {
@@ -811,7 +810,7 @@ class TargetTransformInfoImplCRTPBase : public TargetTransformInfoImplBase {
 
         return TTI::TCC_Basic * (FTy->getNumParams() + 1);
       }
-      return TTI::TCC_Basic * (CS.arg_size() + 1);
+      return TTI::TCC_Basic * (CB->arg_size() + 1);
     }
 
     Type *Ty = U->getType();

diff --git a/llvm/lib/Target/SystemZ/SystemZTargetTransformInfo.cpp b/llvm/lib/Target/SystemZ/SystemZTargetTransformInfo.cpp
@@ -246,8 +246,7 @@ void SystemZTTIImpl::getUnrollingPreferences(Loop *L, ScalarEvolution &SE,
   for (auto &BB : L->blocks())
     for (auto &I : *BB) {
       if (isa<CallInst>(&I) || isa<InvokeInst>(&I)) {
-        ImmutableCallSite CS(&I);
-        if (const Function *F = CS.getCalledFunction()) {
+        if (const Function *F = cast<CallBase>(I).getCalledFunction()) {
           if (isLoweredToCall(F))
             HasCall = true;
           if (F->getIntrinsicID() == Intrinsic::memcpy ||