xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AMDGPU/SIProgramInfo.h (revision 942815c54820783d3d4f7f6faa71ab7919b5f0e5)
1 //===--- SIProgramInfo.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 /// \file
10 /// Defines struct to track resource usage and hardware flags for kernels and
11 /// entry functions.
12 ///
13 //
14 //===----------------------------------------------------------------------===//
15 
16 #ifndef LLVM_LIB_TARGET_AMDGPU_SIPROGRAMINFO_H
17 #define LLVM_LIB_TARGET_AMDGPU_SIPROGRAMINFO_H
18 
19 #include "llvm/IR/CallingConv.h"
20 #include <cstdint>
21 
22 namespace llvm {
23 
24 /// Track resource usage for kernels / entry functions.
25 struct SIProgramInfo {
26     // Fields set in PGM_RSRC1 pm4 packet.
27     uint32_t VGPRBlocks = 0;
28     uint32_t SGPRBlocks = 0;
29     uint32_t Priority = 0;
30     uint32_t FloatMode = 0;
31     uint32_t Priv = 0;
32     uint32_t DX10Clamp = 0;
33     uint32_t DebugMode = 0;
34     uint32_t IEEEMode = 0;
35     uint32_t WgpMode = 0; // GFX10+
36     uint32_t MemOrdered = 0; // GFX10+
37     uint64_t ScratchSize = 0;
38 
39     // Fields set in PGM_RSRC2 pm4 packet.
40     uint32_t LDSBlocks = 0;
41     uint32_t ScratchBlocks = 0;
42 
43     uint64_t ComputePGMRSrc2 = 0;
44     uint64_t ComputePGMRSrc3GFX90A = 0;
45 
46     uint32_t NumVGPR = 0;
47     uint32_t NumArchVGPR = 0;
48     uint32_t NumAccVGPR = 0;
49     uint32_t AccumOffset = 0;
50     uint32_t TgSplit = 0;
51     uint32_t NumSGPR = 0;
52     unsigned SGPRSpill = 0;
53     unsigned VGPRSpill = 0;
54     uint32_t LDSSize = 0;
55     bool FlatUsed = false;
56 
57     // Number of SGPRs that meets number of waves per execution unit request.
58     uint32_t NumSGPRsForWavesPerEU = 0;
59 
60     // Number of VGPRs that meets number of waves per execution unit request.
61     uint32_t NumVGPRsForWavesPerEU = 0;
62 
63     // Final occupancy.
64     uint32_t Occupancy = 0;
65 
66     // Whether there is recursion, dynamic allocas, indirect calls or some other
67     // reason there may be statically unknown stack usage.
68     bool DynamicCallStack = false;
69 
70     // Bonus information for debugging.
71     bool VCCUsed = false;
72 
73     SIProgramInfo() = default;
74 
75     /// Compute the value of the ComputePGMRsrc1 register.
76     uint64_t getComputePGMRSrc1() const;
77     uint64_t getPGMRSrc1(CallingConv::ID CC) const;
78 };
79 
80 } // namespace llvm
81 
82 #endif // LLVM_LIB_TARGET_AMDGPU_SIPROGRAMINFO_H
83