Home | History | Annotate | Download | only in NVPTX
      1 //===-- NVPTXUtilities - Utilities -----------------------------*- C++ -*-====//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // This file contains the declaration of the NVVM specific utility functions.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXUTILITIES_H
     15 #define LLVM_LIB_TARGET_NVPTX_NVPTXUTILITIES_H
     16 
     17 #include "llvm/IR/Function.h"
     18 #include "llvm/IR/GlobalVariable.h"
     19 #include "llvm/IR/IntrinsicInst.h"
     20 #include "llvm/IR/Value.h"
     21 #include <cstdarg>
     22 #include <set>
     23 #include <string>
     24 #include <vector>
     25 
     26 namespace llvm {
     27 
     28 #define NVCL_IMAGE2D_READONLY_FUNCNAME "__is_image2D_readonly"
     29 #define NVCL_IMAGE3D_READONLY_FUNCNAME "__is_image3D_readonly"
     30 
     31 void clearAnnotationCache(const llvm::Module *);
     32 
     33 bool findOneNVVMAnnotation(const llvm::GlobalValue *, std::string, unsigned &);
     34 bool findAllNVVMAnnotation(const llvm::GlobalValue *, std::string,
     35                            std::vector<unsigned> &);
     36 
     37 bool isTexture(const llvm::Value &);
     38 bool isSurface(const llvm::Value &);
     39 bool isSampler(const llvm::Value &);
     40 bool isImage(const llvm::Value &);
     41 bool isImageReadOnly(const llvm::Value &);
     42 bool isImageWriteOnly(const llvm::Value &);
     43 bool isImageReadWrite(const llvm::Value &);
     44 bool isManaged(const llvm::Value &);
     45 
     46 std::string getTextureName(const llvm::Value &);
     47 std::string getSurfaceName(const llvm::Value &);
     48 std::string getSamplerName(const llvm::Value &);
     49 
     50 bool getMaxNTIDx(const llvm::Function &, unsigned &);
     51 bool getMaxNTIDy(const llvm::Function &, unsigned &);
     52 bool getMaxNTIDz(const llvm::Function &, unsigned &);
     53 
     54 bool getReqNTIDx(const llvm::Function &, unsigned &);
     55 bool getReqNTIDy(const llvm::Function &, unsigned &);
     56 bool getReqNTIDz(const llvm::Function &, unsigned &);
     57 
     58 bool getMinCTASm(const llvm::Function &, unsigned &);
     59 bool isKernelFunction(const llvm::Function &);
     60 
     61 bool getAlign(const llvm::Function &, unsigned index, unsigned &);
     62 bool getAlign(const llvm::CallInst &, unsigned index, unsigned &);
     63 
     64 BasicBlock *getParentBlock(Value *v);
     65 Function *getParentFunction(Value *v);
     66 void dumpBlock(Value *v, char *blockName);
     67 Instruction *getInst(Value *base, char *instName);
     68 void dumpInst(Value *base, char *instName);
     69 void dumpInstRec(Value *v, std::set<Instruction *> *visited);
     70 void dumpInstRec(Value *v);
     71 void dumpParent(Value *v);
     72 
     73 }
     74 
     75 #endif
     76