SPIRVPrepareFunctions.cpp File Reference

#include "SPIRVPrepareFunctions.h"
#include "SPIRV.h"
#include "SPIRVBuiltins.h"
#include "SPIRVSubtarget.h"
#include "SPIRVTargetMachine.h"
#include "SPIRVUtils.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/Analysis/TargetTransformInfo.h"
#include "llvm/Analysis/ValueTracking.h"
#include "llvm/CodeGen/IntrinsicLowering.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/InstIterator.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/IntrinsicInst.h"
#include "llvm/IR/Intrinsics.h"
#include "llvm/IR/IntrinsicsSPIRV.h"
#include "llvm/Transforms/Utils/Cloning.h"
#include "llvm/Transforms/Utils/Local.h"
#include "llvm/Transforms/Utils/LowerMemIntrinsics.h"
#include <regex>

Go to the source code of this file.

Functions
	INITIALIZE_PASS (SPIRVPrepareFunctionsLegacy, "spirv-prepare-functions", "SPIRV prepare functions", false, false) static std
static Function *	getOrCreateFunction (Module *M, Type *RetTy, ArrayRef< Type * > ArgTypes, StringRef Name)
static bool	lowerIntrinsicToFunction (IntrinsicInst *Intrinsic, const TargetTransformInfo &TTI)
static std::string	getAnnotation (Value *AnnoVal, Value *OptAnnoVal)
static SmallVector< Metadata * >	parseAnnotation (Value *I, const std::string &Anno, LLVMContext &Ctx, Type *Int32Ty)
static void	lowerPtrAnnotation (IntrinsicInst *II)
static void	lowerFunnelShifts (IntrinsicInst *FSHIntrinsic)
static void	lowerConstrainedFPCmpIntrinsic (ConstrainedFPCmpIntrinsic *ConstrainedCmpIntrinsic, SmallVector< Instruction * > &EraseFromParent)
static void	lowerExpectAssume (IntrinsicInst *II)
static bool	toSpvLifetimeIntrinsic (IntrinsicInst *II, Intrinsic::ID NewID)
static void	lowerConstrainedFmuladd (IntrinsicInst *II, SmallVector< Instruction * > &EraseFromParent)
static void	addFunctionTypeMutation (NamedMDNode *NMD, SmallVector< std::pair< int, Type * > > ChangedTys, StringRef Name, StringRef AsmConstraints="")
static bool	isAbortKHRBuiltin (const Function &F)
static void	rewriteAbortKHRCall (CallInst *CI)
static std::string	fixMultiOutputConstraintString (StringRef Constraints)

Function Documentation

addFunctionTypeMutation()

void addFunctionTypeMutation ( NamedMDNode * NMD, SmallVector< std::pair< int, Type * > > ChangedTys, StringRef Name, StringRef AsmConstraints = "" )

static

Definition at line 522 of file SPIRVPrepareFunctions.cpp.

References llvm::NamedMDNode::addOperand(), llvm::ConstantAsMetadata::get(), llvm::MDNode::get(), llvm::MDString::get(), llvm::ValueAsMetadata::get(), llvm::Module::getContext(), llvm::Type::getInt32Ty(), llvm::Constant::getNullValue(), llvm::NamedMDNode::getParent(), llvm::SmallVectorTemplateBase< T, bool >::push_back(), and llvm::transform().

fixMultiOutputConstraintString()

std::string fixMultiOutputConstraintString ( StringRef Constraints )

static

Definition at line 715 of file SPIRVPrepareFunctions.cpp.

References llvm::SmallVectorTemplateCommon< T, typename >::back(), I, llvm::SmallVectorTemplateCommon< T, typename >::size(), llvm::SplitString(), and starts_with().

getAnnotation()

std::string getAnnotation ( Value * AnnoVal, Value * OptAnnoVal )

static

Definition at line 176 of file SPIRVPrepareFunctions.cpp.

References llvm::CallingConv::C, llvm::dyn_cast(), llvm::dyn_cast_or_null(), E(), llvm::getConstantStringInfo(), I, and llvm::Ref.

Referenced by lowerPtrAnnotation().

getOrCreateFunction()

Function * getOrCreateFunction ( Module * M, Type * RetTy, ArrayRef< Type * > ArgTypes, StringRef Name )

static

Definition at line 97 of file SPIRVPrepareFunctions.cpp.

References llvm::Function::Create(), llvm::GlobalValue::ExternalLinkage, F, llvm::FunctionType::get(), llvm::Function::setCallingConv(), llvm::GlobalValue::setDSOLocal(), and llvm::CallingConv::SPIR_FUNC.

Referenced by lowerFunnelShifts().

INITIALIZE_PASS()

INITIALIZE_PASS	(	SPIRVPrepareFunctionsLegacy	,
		"spirv-prepare-functions"	,
		"SPIRV prepare functions"	,
		false	,
		false	)

Definition at line 85 of file SPIRVPrepareFunctions.cpp.

References assert(), llvm::Value::getName(), II, llvm::replace(), and llvm::StringRef::str().

isAbortKHRBuiltin()

bool isAbortKHRBuiltin ( const Function & F )

static

Definition at line 616 of file SPIRVPrepareFunctions.cpp.

References F, llvm::getOclOrSpirvBuiltinDemangledName(), and llvm::SPIRV::lookupBuiltinNameHelper().

lowerConstrainedFmuladd()

void lowerConstrainedFmuladd ( IntrinsicInst * II, SmallVector< Instruction * > & EraseFromParent )

static

Definition at line 420 of file SPIRVPrepareFunctions.cpp.

References A(), llvm::Add, llvm::cast(), II, Mul, and llvm::SmallVectorTemplateBase< T, bool >::push_back().

lowerConstrainedFPCmpIntrinsic()

void lowerConstrainedFPCmpIntrinsic ( ConstrainedFPCmpIntrinsic * ConstrainedCmpIntrinsic, SmallVector< Instruction * > & EraseFromParent )

static

Definition at line 361 of file SPIRVPrepareFunctions.cpp.

References llvm::dyn_cast(), llvm::CallBase::getArgOperand(), llvm::ConstrainedFPCmpIntrinsic::getPredicate(), LHS, llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::Value::replaceAllUsesWith(), and RHS.

lowerExpectAssume()

void lowerExpectAssume ( IntrinsicInst * II )

static

Definition at line 376 of file SPIRVPrepareFunctions.cpp.

References F, llvm::Intrinsic::getOrInsertDeclaration(), II, and llvm_unreachable.

lowerFunnelShifts()

void lowerFunnelShifts ( IntrinsicInst * FSHIntrinsic )

static

Definition at line 300 of file SPIRVPrepareFunctions.cpp.

References llvm::BitWidth, llvm::BasicBlock::Create(), llvm::IRBuilderBase::CreateLShr(), llvm::IRBuilderBase::CreateOr(), llvm::IRBuilderBase::CreateRet(), llvm::IRBuilderBase::CreateShl(), llvm::IRBuilderBase::CreateSub(), llvm::IRBuilderBase::CreateURem(), llvm::IRBuilderBase::CreateVectorSplat(), llvm::dyn_cast(), llvm::Function::empty(), llvm::Function::getArg(), llvm::VectorType::getElementType(), llvm::CallBase::getFunctionType(), llvm::IRBuilderBase::getInt(), llvm::IntrinsicInst::getIntrinsicID(), llvm::Instruction::getModule(), llvm::FixedVectorType::getNumElements(), getOrCreateFunction(), llvm::Function::getReturnType(), and llvm::CallBase::setCalledFunction().

lowerIntrinsicToFunction()

bool lowerIntrinsicToFunction ( IntrinsicInst * Intrinsic, const TargetTransformInfo & TTI )

static

Definition at line 111 of file SPIRVPrepareFunctions.cpp.

References assert(), llvm::cast(), llvm::BasicBlock::Create(), llvm::IRBuilderBase::CreateIntrinsic(), llvm::IRBuilderBase::CreateMemSet(), llvm::IRBuilderBase::CreateRet(), llvm::IRBuilderBase::CreateRetVoid(), llvm::dyn_cast(), llvm::expandMemSetAsLoop(), F, llvm::isa(), llvm::IntrinsicLowering::LowerIntrinsicCall(), and llvm::Value::setName().

lowerPtrAnnotation()

void lowerPtrAnnotation ( IntrinsicInst * II )

static

Definition at line 264 of file SPIRVPrepareFunctions.cpp.

References llvm::IRBuilderBase::CreateIntrinsic(), llvm::dyn_cast(), llvm::ConstantAsMetadata::get(), llvm::MDNode::get(), llvm::MDString::get(), llvm::MetadataAsValue::get(), getAnnotation(), llvm::Type::getInt32Ty(), llvm::Value::getType(), II, llvm::Int32Ty, parseAnnotation(), llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::IRBuilderBase::SetInsertPoint(), and llvm::SmallVectorTemplateCommon< T, typename >::size().

parseAnnotation()

SmallVector< Metadata * > parseAnnotation ( Value * I, const std::string & Anno, LLVMContext & Ctx, Type * Int32Ty )

static

Definition at line 211 of file SPIRVPrepareFunctions.cpp.

References llvm::ConstantAsMetadata::get(), llvm::MDNode::get(), llvm::MDString::get(), I, llvm::Int32Ty, llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::SmallVectorTemplateCommon< T, typename >::size(), and llvm::to_integer().

Referenced by lowerPtrAnnotation().

rewriteAbortKHRCall()

void rewriteAbortKHRCall ( CallInst * CI )

static

Definition at line 635 of file SPIRVPrepareFunctions.cpp.

References B(), llvm::changeToUnreachable(), llvm::CallBase::getArgOperand(), llvm::CallBase::getParamByValType(), llvm::Value::getType(), and llvm::CallBase::isByValArgument().

toSpvLifetimeIntrinsic()

bool toSpvLifetimeIntrinsic ( IntrinsicInst * II, Intrinsic::ID NewID )

static

Definition at line 400 of file SPIRVPrepareFunctions.cpp.

References llvm::cast(), II, llvm::isa(), and Size.