1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#include "AdaptorCommon/ImplicitArgs.hpp"
#include "Compiler/Optimizer/OpenCLPasses/OpenCLPrintf/OpenCLPrintfAnalysis.hpp"
#include "Compiler/IGCPassSupport.h"
#include "common/LLVMWarningsPush.hpp"
#include <llvm/IR/Module.h>
#include <llvm/IR/Function.h>
#include <llvm/ADT/StringRef.h>
#include "common/LLVMWarningsPop.hpp"
using namespace llvm;
using namespace IGC;
using namespace IGC::IGCMD;
// Register pass to igc-opt
#define PASS_FLAG "igc-opencl-printf-analysis"
#define PASS_DESCRIPTION "Analyzes OpenCL printf calls"
#define PASS_CFG_ONLY false
#define PASS_ANALYSIS false
IGC_INITIALIZE_PASS_BEGIN(OpenCLPrintfAnalysis, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
IGC_INITIALIZE_PASS_DEPENDENCY(MetaDataUtilsWrapper)
IGC_INITIALIZE_PASS_END(OpenCLPrintfAnalysis, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
char OpenCLPrintfAnalysis::ID = 0;
OpenCLPrintfAnalysis::OpenCLPrintfAnalysis() : ModulePass(ID)
{
initializeOpenCLPrintfAnalysisPass(*PassRegistry::getPassRegistry());
}
//TODO: move to a common place
const StringRef OpenCLPrintfAnalysis::OPENCL_PRINTF_FUNCTION_NAME = "printf";
bool OpenCLPrintfAnalysis::runOnModule(Module& M)
{
m_pMDUtils = getAnalysis<MetaDataUtilsWrapper>().getMetaDataUtils();
visit(M);
bool changed = false;
if (m_hasPrintfs.size())
{
for (Function& func : M.getFunctionList())
{
if (!func.isDeclaration() &&
m_hasPrintfs.find(&func) != m_hasPrintfs.end())
{
addPrintfBufferArgs(func);
changed = true;
}
}
}
// Update LLVM metadata based on IGC MetadataUtils
if (changed)
m_pMDUtils->save(M.getContext());
return m_hasPrintfs.size();
}
void OpenCLPrintfAnalysis::visitCallInst(llvm::CallInst& callInst)
{
Function* pF = callInst.getParent()->getParent();
if (!callInst.getCalledFunction() || m_hasPrintfs.find(pF)!=m_hasPrintfs.end())
{
return;
}
StringRef funcName = callInst.getCalledFunction()->getName();
bool hasPrintf = (funcName == OpenCLPrintfAnalysis::OPENCL_PRINTF_FUNCTION_NAME);
if (hasPrintf)
{
m_hasPrintfs.insert(pF);
}
}
void OpenCLPrintfAnalysis::addPrintfBufferArgs(Function& F)
{
SmallVector<ImplicitArg::ArgType, 1> implicitArgs;
implicitArgs.push_back(ImplicitArg::PRINTF_BUFFER);
ImplicitArgs::addImplicitArgs(F, implicitArgs, m_pMDUtils);
}
|