llvm/lib/Target/NVPTX/NVPTXUtilities.h

   1 //===-- NVPTXUtilities - Utilities -----------------------------*- C++ -*-====//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file contains the declaration of the NVVM specific utility functions.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXUTILITIES_H
  14 #define LLVM_LIB_TARGET_NVPTX_NVPTXUTILITIES_H
  15
  16 #include "llvm/CodeGen/ValueTypes.h"
  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 "llvm/Support/Alignment.h"
  22 #include <cstdarg>
  23 #include <set>
  24 #include <string>
  25 #include <vector>
  26
  27 namespace llvm {
  28
  29 class TargetMachine;
  30
  31 void clearAnnotationCache(const Module *);
  32
  33 bool findOneNVVMAnnotation(const GlobalValue *, const std::string &,
  34                            unsigned &);
  35 bool findAllNVVMAnnotation(const GlobalValue *, const std::string &,
  36                            std::vector<unsigned> &);
  37
  38 bool isTexture(const Value &);
  39 bool isSurface(const Value &);
  40 bool isSampler(const Value &);
  41 bool isImage(const Value &);
  42 bool isImageReadOnly(const Value &);
  43 bool isImageWriteOnly(const Value &);
  44 bool isImageReadWrite(const Value &);
  45 bool isManaged(const Value &);
  46
  47 std::string getTextureName(const Value &);
  48 std::string getSurfaceName(const Value &);
  49 std::string getSamplerName(const Value &);
  50
  51 std::optional<unsigned> getMaxNTIDx(const Function &);
  52 std::optional<unsigned> getMaxNTIDy(const Function &);
  53 std::optional<unsigned> getMaxNTIDz(const Function &);
  54 std::optional<unsigned> getMaxNTID(const Function &F);
  55
  56 std::optional<unsigned> getReqNTIDx(const Function &);
  57 std::optional<unsigned> getReqNTIDy(const Function &);
  58 std::optional<unsigned> getReqNTIDz(const Function &);
  59 std::optional<unsigned> getReqNTID(const Function &);
  60
  61 bool getMaxClusterRank(const Function &, unsigned &);
  62 bool getMinCTASm(const Function &, unsigned &);
  63 bool getMaxNReg(const Function &, unsigned &);
  64 bool isKernelFunction(const Function &);
  65 bool isParamGridConstant(const Value &);
  66
  67 MaybeAlign getAlign(const Function &, unsigned);
  68 MaybeAlign getAlign(const CallInst &, unsigned);
  69 Function *getMaybeBitcastedCallee(const CallBase *CB);
  70
  71 // PTX ABI requires all scalar argument/return values to have
  72 // bit-size as a power of two of at least 32 bits.
  73 inline unsigned promoteScalarArgumentSize(unsigned size) {
  74   if (size <= 32)
  75     return 32;
  76   else if (size <= 64)
  77     return 64;
  78   else
  79     return size;
  80 }
  81
  82 bool shouldEmitPTXNoReturn(const Value *V, const TargetMachine &TM);
  83
  84 bool Isv2x16VT(EVT VT);
  85 }
  86
  87 #endif