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/IR/Function.h"
17 #include "llvm/IR/GlobalVariable.h"
18 #include "llvm/IR/IntrinsicInst.h"
19 #include "llvm/IR/Value.h"
20 #include <cstdarg>
21 #include <set>
22 #include <string>
23 #include <vector>
24
25 namespace llvm {
26
27 class TargetMachine;
28
29 void clearAnnotationCache(const Module *);
30
31 bool findOneNVVMAnnotation(const GlobalValue *, const std::string &,
32 unsigned &);
33 bool findAllNVVMAnnotation(const GlobalValue *, const std::string &,
34 std::vector<unsigned> &);
35
36 bool isTexture(const Value &);
37 bool isSurface(const Value &);
38 bool isSampler(const Value &);
39 bool isImage(const Value &);
40 bool isImageReadOnly(const Value &);
41 bool isImageWriteOnly(const Value &);
42 bool isImageReadWrite(const Value &);
43 bool isManaged(const Value &);
44
45 std::string getTextureName(const Value &);
46 std::string getSurfaceName(const Value &);
47 std::string getSamplerName(const Value &);
48
49 bool getMaxNTIDx(const Function &, unsigned &);
50 bool getMaxNTIDy(const Function &, unsigned &);
51 bool getMaxNTIDz(const Function &, unsigned &);
52
53 bool getReqNTIDx(const Function &, unsigned &);
54 bool getReqNTIDy(const Function &, unsigned &);
55 bool getReqNTIDz(const Function &, unsigned &);
56
57 bool getMinCTASm(const Function &, unsigned &);
58 bool getMaxNReg(const Function &, unsigned &);
59 bool isKernelFunction(const Function &);
60
61 bool getAlign(const Function &, unsigned index, unsigned &);
62 bool getAlign(const CallInst &, unsigned index, unsigned &);
63 Function *getMaybeBitcastedCallee(const CallBase *CB);
64
65 // PTX ABI requires all scalar argument/return values to have
66 // bit-size as a power of two of at least 32 bits.
promoteScalarArgumentSize(unsigned size)67 inline unsigned promoteScalarArgumentSize(unsigned size) {
68 if (size <= 32)
69 return 32;
70 else if (size <= 64)
71 return 64;
72 else
73 return size;
74 }
75
76 bool shouldEmitPTXNoReturn(const Value *V, const TargetMachine &TM);
77 }
78
79 #endif
80