[llvm] 744eaa7 - [CallSite removal][AMDGPU] Use CallBase instead of CallSite in AMDGPUFixFunctionBitcasts. NFC

[Craig Topper via llvm-commits]

Author: Craig Topper
Date: 2020-04-19T15:21:02-07:00
New Revision: 744eaa7a3fc13b2e235216fa208345387f9b67e4

URL: https://github.com/llvm/llvm-project/commit/744eaa7a3fc13b2e235216fa208345387f9b67e4
DIFF: https://github.com/llvm/llvm-project/commit/744eaa7a3fc13b2e235216fa208345387f9b67e4.diff

LOG: [CallSite removal][AMDGPU] Use CallBase instead of CallSite in AMDGPUFixFunctionBitcasts. NFC

Added: 
    

Modified: 
    llvm/lib/Target/AMDGPU/AMDGPUFixFunctionBitcasts.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/AMDGPU/AMDGPUFixFunctionBitcasts.cpp b/llvm/lib/Target/AMDGPU/AMDGPUFixFunctionBitcasts.cpp
index 2c23a163e70b..5fbb00908e7e 100644
--- a/llvm/lib/Target/AMDGPU/AMDGPUFixFunctionBitcasts.cpp
+++ b/llvm/lib/Target/AMDGPU/AMDGPUFixFunctionBitcasts.cpp
@@ -31,13 +31,12 @@ class AMDGPUFixFunctionBitcasts final
   bool Modified;
 
 public:
-  void visitCallSite(CallSite CS) {
-    if (CS.getCalledFunction())
+  void visitCallBase(CallBase &CB) {
+    if (CB.getCalledFunction())
       return;
-    auto Callee = dyn_cast<Function>(CS.getCalledValue()->stripPointerCasts());
-    if (Callee &&
-        isLegalToPromote(*cast<CallBase>(CS.getInstruction()), Callee)) {
-      promoteCall(*cast<CallBase>(CS.getInstruction()), Callee);
+    auto *Callee = dyn_cast<Function>(CB.getCalledValue()->stripPointerCasts());
+    if (Callee && isLegalToPromote(CB, Callee)) {
+      promoteCall(CB, Callee);
       Modified = true;
     }
   }