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