NVVMProperties.cpp File Reference

#include "NVVMProperties.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/IR/Argument.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"
#include "llvm/IR/GlobalVariable.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/Metadata.h"
#include "llvm/IR/Module.h"
#include "llvm/Support/ModRef.h"
#include "llvm/Support/Mutex.h"
#include "llvm/Support/NVVMAttributes.h"
#include <functional>
#include <map>
#include <mutex>
#include <numeric>
#include <string>
#include <vector>

Go to the source code of this file.

Namespaces
namespace	NVVMMetadata

Functions
static AnnotationCache &	getAnnotationCache ()
constexpr StringLiteral	NVVMMetadata::Texture ("texture")
constexpr StringLiteral	NVVMMetadata::Surface ("surface")
constexpr StringLiteral	NVVMMetadata::Sampler ("sampler")
constexpr StringLiteral	NVVMMetadata::ReadOnlyImage ("rdoimage")
constexpr StringLiteral	NVVMMetadata::WriteOnlyImage ("wroimage")
constexpr StringLiteral	NVVMMetadata::ReadWriteImage ("rdwrimage")
constexpr StringLiteral	NVVMMetadata::Managed ("managed")
static void	cacheAnnotationFromMD (const MDNode *MetadataNode, AnnotationValues &RetVal)
static void	cacheAnnotationFromMD (const Module *M, const GlobalValue *GV)
static std::optional< unsigned >	findOneNVVMAnnotation (const GlobalValue *GV, StringRef Prop)
static bool	findAllNVVMAnnotation (const GlobalValue *GV, StringRef Prop, std::vector< unsigned > &RetVal)
static bool	globalHasNVVMAnnotation (const Value &V, StringRef Prop)
static bool	argHasNVVMAnnotation (const Value &Val, StringRef Annotation)
static std::optional< unsigned >	getFnAttrParsedInt (const Function &F, StringRef Attr)
static SmallVector< unsigned, 3 >	getFnAttrParsedVector (const Function &F, StringRef Attr)
static std::optional< uint64_t >	getVectorProduct (ArrayRef< unsigned > V)

Function Documentation

argHasNVVMAnnotation()

bool argHasNVVMAnnotation ( const Value & Val, StringRef Annotation )

static

Definition at line 166 of file NVVMProperties.cpp.

References llvm::dyn_cast(), findAllNVVMAnnotation(), and llvm::is_contained().

Referenced by llvm::getPTXOpaqueType().

cacheAnnotationFromMD() [1/2]

void cacheAnnotationFromMD ( const MDNode * MetadataNode, AnnotationValues & RetVal )

static

Definition at line 71 of file NVVMProperties.cpp.

References assert(), llvm::dyn_cast(), llvm::mdconst::dyn_extract(), E(), getAnnotationCache(), llvm::MDNode::getNumOperands(), llvm::MDNode::getOperand(), llvm::MDString::getString(), I, llvm::InnerAnalysisManagerProxy< AnalysisManagerT, IRUnitT, ExtraArgTs... >::Key, llvm_unreachable, and llvm::StringRef::str().

Referenced by cacheAnnotationFromMD(), findAllNVVMAnnotation(), and findOneNVVMAnnotation().

cacheAnnotationFromMD() [2/2]

void cacheAnnotationFromMD ( const Module * M, const GlobalValue * GV )

static

Definition at line 94 of file NVVMProperties.cpp.

References cacheAnnotationFromMD(), llvm::mdconst::dyn_extract_or_null(), E(), getAnnotationCache(), llvm::NamedMDNode::getNumOperands(), llvm::MDNode::getOperand(), llvm::NamedMDNode::getOperand(), and I.

findAllNVVMAnnotation()

bool findAllNVVMAnnotation ( const GlobalValue * GV, StringRef Prop, std::vector< unsigned > & RetVal )

static

Definition at line 137 of file NVVMProperties.cpp.

References cacheAnnotationFromMD(), getAnnotationCache(), llvm::GlobalValue::getParent(), and llvm::StringRef::str().

Referenced by argHasNVVMAnnotation().

findOneNVVMAnnotation()

std::optional< unsigned > findOneNVVMAnnotation ( const GlobalValue * GV, StringRef Prop )

static

Definition at line 119 of file NVVMProperties.cpp.

References cacheAnnotationFromMD(), getAnnotationCache(), llvm::GlobalValue::getParent(), and llvm::StringRef::str().

Referenced by llvm::getPTXOpaqueType(), and globalHasNVVMAnnotation().

getAnnotationCache()

AnnotationCache & getAnnotationCache ( )

static

Definition at line 48 of file NVVMProperties.cpp.

Referenced by cacheAnnotationFromMD(), cacheAnnotationFromMD(), llvm::clearAnnotationCache(), findAllNVVMAnnotation(), and findOneNVVMAnnotation().

getFnAttrParsedInt()

std::optional< unsigned > getFnAttrParsedInt ( const Function & F, StringRef Attr )

static

Definition at line 176 of file NVVMProperties.cpp.

References F.

Referenced by llvm::getMaxClusterRank(), llvm::getMaxNReg(), and llvm::getMinCTASm().

getFnAttrParsedVector()

SmallVector< unsigned, 3 > getFnAttrParsedVector ( const Function & F, StringRef Attr )

static

Definition at line 183 of file NVVMProperties.cpp.

References llvm::StringRef::empty(), F, llvm::First, I, and llvm::StringRef::split().

Referenced by llvm::getClusterDim(), llvm::getMaxNTID(), and llvm::getReqNTID().

getVectorProduct()

std::optional< uint64_t > getVectorProduct ( ArrayRef< unsigned > V )

static

Definition at line 205 of file NVVMProperties.cpp.

Referenced by llvm::getOverallClusterRank(), llvm::getOverallMaxNTID(), and llvm::getOverallReqNTID().

globalHasNVVMAnnotation()

bool globalHasNVVMAnnotation ( const Value & V, StringRef Prop )

static

Definition at line 156 of file NVVMProperties.cpp.

References assert(), llvm::dyn_cast(), and findOneNVVMAnnotation().

Referenced by llvm::isManaged().