xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AMDGPU/AMDGPUMachineFunction.h (revision 9dba64be9536c28e4800e06512b7f29b43ade345)
1 //===-- AMDGPUMachineFunctionInfo.h -------------------------------*- 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 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINEFUNCTION_H
10 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINEFUNCTION_H
11 
12 #include "llvm/ADT/DenseMap.h"
13 #include "llvm/CodeGen/MachineFunction.h"
14 
15 namespace llvm {
16 
17 class GCNSubtarget;
18 
19 class AMDGPUMachineFunction : public MachineFunctionInfo {
20   /// A map to keep track of local memory objects and their offsets within the
21   /// local memory space.
22   SmallDenseMap<const GlobalValue *, unsigned, 4> LocalMemoryObjects;
23 
24 protected:
25   uint64_t ExplicitKernArgSize; // Cache for this.
26   Align MaxKernArgAlign;        // Cache for this.
27 
28   /// Number of bytes in the LDS that are being used.
29   unsigned LDSSize;
30 
31   // Kernels + shaders. i.e. functions called by the driver and not called
32   // by other functions.
33   bool IsEntryFunction;
34 
35   bool NoSignedZerosFPMath;
36 
37   // Function may be memory bound.
38   bool MemoryBound;
39 
40   // Kernel may need limited waves per EU for better performance.
41   bool WaveLimiter;
42 
43 public:
44   AMDGPUMachineFunction(const MachineFunction &MF);
45 
46   uint64_t getExplicitKernArgSize() const {
47     return ExplicitKernArgSize;
48   }
49 
50   unsigned getMaxKernArgAlign() const { return MaxKernArgAlign.value(); }
51 
52   unsigned getLDSSize() const {
53     return LDSSize;
54   }
55 
56   bool isEntryFunction() const {
57     return IsEntryFunction;
58   }
59 
60   bool hasNoSignedZerosFPMath() const {
61     return NoSignedZerosFPMath;
62   }
63 
64   bool isMemoryBound() const {
65     return MemoryBound;
66   }
67 
68   bool needsWaveLimiter() const {
69     return WaveLimiter;
70   }
71 
72   unsigned allocateLDSGlobal(const DataLayout &DL, const GlobalValue &GV);
73 };
74 
75 }
76 #endif
77