Commit 57aa7f7e authored by Bill Wendling's avatar Bill Wendling
Browse files

Merging r182297: 

------------------------------------------------------------------------
r182297 | jholewinski | 2013-05-20 09:42:16 -0700 (Mon, 20 May 2013) | 1 line

[NVPTX] Add programmatic interface to NVVMReflect pass
------------------------------------------------------------------------

llvm-svn: 182827
parent b0bb2e99

llvm/lib/Target/NVPTX/NVPTX.h

+3 −0
Original line number Diff line number Diff line
@@ -16,6 +16,7 @@ 
#define LLVM_TARGET_NVPTX_H



#include "MCTargetDesc/NVPTXBaseInfo.h"

#include "llvm/ADT/StringMap.h"

#include "llvm/IR/Module.h"

#include "llvm/IR/Value.h"

#include "llvm/Support/ErrorHandling.h"
@@ -63,6 +64,8 @@ FunctionPass *createLowerStructArgsPass(NVPTXTargetMachine &); 
FunctionPass *createNVPTXReMatPass(NVPTXTargetMachine &);

FunctionPass *createNVPTXReMatBlockPass(NVPTXTargetMachine &);

ModulePass *createGenericToNVVMPass();

ModulePass *createNVVMReflectPass();

ModulePass *createNVVMReflectPass(const StringMap<int>& Mapping);



bool isImageOrSamplerVal(const Value *, const Module *);

llvm/lib/Target/NVPTX/NVVMReflect.cpp

+21 −3
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ 
//

//===----------------------------------------------------------------------===//



#include "NVPTX.h"

#include "llvm/ADT/DenseMap.h"

#include "llvm/ADT/SmallVector.h"

#include "llvm/ADT/StringMap.h"
@@ -40,7 +41,7 @@ using namespace llvm; 
namespace llvm { void initializeNVVMReflectPass(PassRegistry &); }



namespace {

class LLVM_LIBRARY_VISIBILITY NVVMReflect : public ModulePass {

class NVVMReflect : public ModulePass {

private:

  StringMap<int> VarMap;

  typedef DenseMap<std::string, int>::iterator VarMapIter;
@@ -48,9 +49,18 @@ private: 


public:

  static char ID;

  NVVMReflect() : ModulePass(ID) {

  NVVMReflect() : ModulePass(ID), ReflectFunction(0) {

    initializeNVVMReflectPass(*PassRegistry::getPassRegistry());

    VarMap.clear();

    ReflectFunction = 0;

  }



  NVVMReflect(const StringMap<int> &Mapping)

  : ModulePass(ID), ReflectFunction(0) {

    initializeNVVMReflectPass(*PassRegistry::getPassRegistry());

    for (StringMap<int>::const_iterator I = Mapping.begin(), E = Mapping.end();

         I != E; ++I) {

      VarMap[(*I).getKey()] = (*I).getValue();

    }

  }



  void getAnalysisUsage(AnalysisUsage &AU) const { AU.setPreservesAll(); }
@@ -60,6 +70,14 @@ public: 
};

}



ModulePass *llvm::createNVVMReflectPass() {

  return new NVVMReflect();

}



ModulePass *llvm::createNVVMReflectPass(const StringMap<int>& Mapping) {

  return new NVVMReflect(Mapping);

}



static cl::opt<bool>

NVVMReflectEnabled("nvvm-reflect-enable", cl::init(true),

                   cl::desc("NVVM reflection, enabled by default"));