| //===-- AMDGPUFixFunctionBitcasts.cpp - Fix function bitcasts -------------===// |
| // |
| // 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 |
| // |
| //===----------------------------------------------------------------------===// |
| /// |
| /// \file |
| /// Promote indirect (bitcast) calls to direct calls when they are statically |
| /// known to be direct. Required when InstCombine is not run (e.g. at OptNone) |
| /// because AMDGPU does not support indirect calls. |
| /// |
| //===----------------------------------------------------------------------===// |
| |
| #include "AMDGPU.h" |
| #include "llvm/IR/InstVisitor.h" |
| #include "llvm/Pass.h" |
| #include "llvm/Transforms/Utils/CallPromotionUtils.h" |
| |
| using namespace llvm; |
| |
| #define DEBUG_TYPE "amdgpu-fix-function-bitcasts" |
| |
| namespace { |
| class AMDGPUFixFunctionBitcasts final |
| : public ModulePass, |
| public InstVisitor<AMDGPUFixFunctionBitcasts> { |
| |
| bool runOnModule(Module &M) override; |
| |
| bool Modified; |
| |
| public: |
| void visitCallBase(CallBase &CB) { |
| if (CB.getCalledFunction()) |
| return; |
| auto *Callee = |
| dyn_cast<Function>(CB.getCalledOperand()->stripPointerCasts()); |
| if (Callee && isLegalToPromote(CB, Callee)) { |
| promoteCall(CB, Callee); |
| Modified = true; |
| } |
| } |
| |
| static char ID; |
| AMDGPUFixFunctionBitcasts() : ModulePass(ID) {} |
| }; |
| } // End anonymous namespace |
| |
| char AMDGPUFixFunctionBitcasts::ID = 0; |
| char &llvm::AMDGPUFixFunctionBitcastsID = AMDGPUFixFunctionBitcasts::ID; |
| INITIALIZE_PASS(AMDGPUFixFunctionBitcasts, DEBUG_TYPE, |
| "Fix function bitcasts for AMDGPU", false, false) |
| |
| ModulePass *llvm::createAMDGPUFixFunctionBitcastsPass() { |
| return new AMDGPUFixFunctionBitcasts(); |
| } |
| |
| bool AMDGPUFixFunctionBitcasts::runOnModule(Module &M) { |
| Modified = false; |
| visit(M); |
| return Modified; |
| } |