LLVM  6.0.0svn
Functions | Variables
AMDGPUOpenCLImageTypeLoweringPass.cpp File Reference

This pass resolves calls to OpenCL image attribute, image resource ID and sampler resource ID getter functions. More...

#include "AMDGPU.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h"
#include "llvm/IR/Argument.h"
#include "llvm/IR/DerivedTypes.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/Instruction.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/Metadata.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Type.h"
#include "llvm/IR/Use.h"
#include "llvm/IR/User.h"
#include "llvm/Pass.h"
#include "llvm/Support/Casting.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Transforms/Utils/Cloning.h"
#include "llvm/Transforms/Utils/ValueMapper.h"
#include <cassert>
#include <cstddef>
#include <cstdint>
#include <tuple>
Include dependency graph for AMDGPUOpenCLImageTypeLoweringPass.cpp:

Go to the source code of this file.

Functions

static bool IsImageType (StringRef TypeString)
 
static bool IsSamplerType (StringRef TypeString)
 
static FunctionGetFunctionFromMDNode (MDNode *Node)
 
static StringRef AccessQualFromMD (MDNode *KernelMDNode, unsigned ArgIdx)
 
static StringRef ArgTypeFromMD (MDNode *KernelMDNode, unsigned ArgIdx)
 
static MDVector GetArgMD (MDNode *KernelMDNode, unsigned OpIdx)
 
static void PushArgMD (KernelArgMD &MD, const MDVector &V)
 

Variables

static StringRef GetImageSizeFunc = "llvm.OpenCL.image.get.size"
 
static StringRef GetImageFormatFunc = "llvm.OpenCL.image.get.format"
 
static StringRef GetImageResourceIDFunc = "llvm.OpenCL.image.get.resource.id"
 
static StringRef GetSamplerResourceIDFunc
 
static StringRef ImageSizeArgMDType = "__llvm_image_size"
 
static StringRef ImageFormatArgMDType = "__llvm_image_format"
 
static StringRef KernelsMDNodeName = "opencl.kernels"
 
static StringRef KernelArgMDNodeNames []
 
static const unsigned NumKernelArgMDNodes = 5
 

Detailed Description

This pass resolves calls to OpenCL image attribute, image resource ID and sampler resource ID getter functions.

Image attributes (size and format) are expected to be passed to the kernel as kernel arguments immediately following the image argument itself, therefore this pass adds image size and format arguments to the kernel functions in the module. The kernel functions with image arguments are re-created using the new signature. The new arguments are added to the kernel metadata with kernel_arg_type set to "image_size" or "image_format". Note: this pass may invalidate pointers to functions.

Resource IDs of read-only images, write-only images and samplers are defined to be their index among the kernel arguments of the same type and access qualifier.

Definition in file AMDGPUOpenCLImageTypeLoweringPass.cpp.

Function Documentation

◆ AccessQualFromMD()

static StringRef AccessQualFromMD ( MDNode KernelMDNode,
unsigned  ArgIdx 
)
static

Definition at line 125 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

References llvm::MDNode::getOperand().

Referenced by PushArgMD().

◆ ArgTypeFromMD()

static StringRef ArgTypeFromMD ( MDNode KernelMDNode,
unsigned  ArgIdx 
)
static

Definition at line 131 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

References llvm::MDNode::getOperand().

Referenced by PushArgMD().

◆ GetArgMD()

static MDVector GetArgMD ( MDNode KernelMDNode,
unsigned  OpIdx 
)
static

Definition at line 137 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

References llvm::MDNode::getOperand(), and NumKernelArgMDNodes.

Referenced by PushArgMD().

◆ GetFunctionFromMDNode()

static Function* GetFunctionFromMDNode ( MDNode Node)
static

◆ IsImageType()

static bool IsImageType ( StringRef  TypeString)
inlinestatic

Definition at line 83 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

Referenced by PushArgMD().

◆ IsSamplerType()

static bool IsSamplerType ( StringRef  TypeString)
inlinestatic

Definition at line 88 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

Referenced by PushArgMD().

◆ PushArgMD()

static void PushArgMD ( KernelArgMD &  MD,
const MDVector &  V 
)
static

Variable Documentation

◆ GetImageFormatFunc

StringRef GetImageFormatFunc = "llvm.OpenCL.image.get.format"
static

Definition at line 56 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ GetImageResourceIDFunc

StringRef GetImageResourceIDFunc = "llvm.OpenCL.image.get.resource.id"
static

Definition at line 57 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ GetImageSizeFunc

StringRef GetImageSizeFunc = "llvm.OpenCL.image.get.size"
static

Definition at line 55 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ GetSamplerResourceIDFunc

StringRef GetSamplerResourceIDFunc
static
Initial value:
=
"llvm.OpenCL.sampler.get.resource.id"

Definition at line 58 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ ImageFormatArgMDType

StringRef ImageFormatArgMDType = "__llvm_image_format"
static

Definition at line 62 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ ImageSizeArgMDType

StringRef ImageSizeArgMDType = "__llvm_image_size"
static

Definition at line 61 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ KernelArgMDNodeNames

StringRef KernelArgMDNodeNames[]
static
Initial value:
= {
"kernel_arg_addr_space",
"kernel_arg_access_qual",
"kernel_arg_type",
"kernel_arg_base_type",
"kernel_arg_type_qual"}

Definition at line 65 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ KernelsMDNodeName

StringRef KernelsMDNodeName = "opencl.kernels"
static

Definition at line 64 of file AMDGPUOpenCLImageTypeLoweringPass.cpp.

◆ NumKernelArgMDNodes

const unsigned NumKernelArgMDNodes = 5
static