1 /*
2 * Copyright © 2019 Valve Corporation.
3 *
4 * SPDX-License-Identifier: MIT
5 */
6
7 #ifndef RADV_SHADER_ARGS_H
8 #define RADV_SHADER_ARGS_H
9
10 #include "compiler/shader_enums.h"
11 #include "util/list.h"
12 #include "util/macros.h"
13 #include "ac_shader_args.h"
14 #include "amd_family.h"
15 #include "radv_constants.h"
16
17 enum radv_ud_index {
18 AC_UD_SCRATCH_RING_OFFSETS = 0,
19 AC_UD_PUSH_CONSTANTS = 1,
20 AC_UD_INLINE_PUSH_CONSTANTS = 2,
21 AC_UD_INDIRECT_DESCRIPTOR_SETS = 3,
22 AC_UD_VIEW_INDEX = 4,
23 AC_UD_STREAMOUT_BUFFERS = 5,
24 AC_UD_STREAMOUT_STATE = 6,
25 AC_UD_TASK_STATE = 7,
26 AC_UD_NGG_CULLING_SETTINGS = 8,
27 AC_UD_NGG_VIEWPORT = 9,
28 AC_UD_NGG_LDS_LAYOUT = 10,
29 AC_UD_NGG_STATE = 11,
30 AC_UD_VGT_ESGS_RING_ITEMSIZE = 12,
31 AC_UD_FORCE_VRS_RATES = 13,
32 AC_UD_TASK_RING_ENTRY = 14,
33 AC_UD_NEXT_STAGE_PC = 15,
34 AC_UD_EPILOG_PC = 16,
35 AC_UD_SHADER_START = 17,
36 AC_UD_VS_VERTEX_BUFFERS = AC_UD_SHADER_START,
37 AC_UD_VS_BASE_VERTEX_START_INSTANCE,
38 AC_UD_VS_PROLOG_INPUTS,
39 AC_UD_VS_MAX_UD,
40 AC_UD_PS_STATE,
41 AC_UD_PS_MAX_UD,
42 AC_UD_CS_GRID_SIZE = AC_UD_SHADER_START,
43 AC_UD_CS_SBT_DESCRIPTORS,
44 AC_UD_CS_RAY_LAUNCH_SIZE_ADDR,
45 AC_UD_CS_RAY_DYNAMIC_CALLABLE_STACK_BASE,
46 AC_UD_CS_TRAVERSAL_SHADER_ADDR,
47 AC_UD_CS_TASK_RING_OFFSETS,
48 AC_UD_CS_TASK_DRAW_ID,
49 AC_UD_CS_TASK_IB,
50 AC_UD_CS_MAX_UD,
51 AC_UD_GS_MAX_UD,
52 AC_UD_TCS_OFFCHIP_LAYOUT = AC_UD_VS_MAX_UD,
53 AC_UD_TCS_MAX_UD,
54 /* We might not know the previous stage when compiling a geometry shader, so we just
55 * declare both TES and VS user SGPRs.
56 */
57 AC_UD_TES_MAX_UD = AC_UD_TCS_MAX_UD,
58 AC_UD_MAX_UD = AC_UD_CS_MAX_UD,
59 };
60
61 struct radv_userdata_info {
62 int8_t sgpr_idx;
63 uint8_t num_sgprs;
64 };
65
66 struct radv_userdata_locations {
67 struct radv_userdata_info descriptor_sets[MAX_SETS];
68 struct radv_userdata_info shader_data[AC_UD_MAX_UD];
69 uint32_t descriptor_sets_enabled;
70 };
71
72 struct radv_shader_args {
73 struct ac_shader_args ac;
74
75 struct ac_arg descriptor_sets[MAX_SETS];
76
77 /* Streamout */
78 struct ac_arg streamout_buffers;
79 struct ac_arg streamout_state; /* GFX12+ */
80
81 /* Task */
82 struct ac_arg task_state;
83 /* User data 2/3. same as ring_offsets but for task shaders. */
84 struct ac_arg task_ring_offsets;
85
86 /* NGG */
87 struct ac_arg ngg_state;
88 struct ac_arg ngg_lds_layout;
89
90 /* NGG GS */
91 struct ac_arg ngg_culling_settings;
92 struct ac_arg ngg_viewport_scale[2];
93 struct ac_arg ngg_viewport_translate[2];
94
95 /* Fragment shaders */
96 struct ac_arg ps_state;
97
98 struct ac_arg prolog_inputs;
99 struct ac_arg vs_inputs[MAX_VERTEX_ATTRIBS];
100
101 /* PS epilogs */
102 struct ac_arg colors[MAX_RTS];
103 struct ac_arg depth;
104 struct ac_arg stencil;
105 struct ac_arg sample_mask;
106
107 /* TCS */
108 /* # [0:6] = the number of tessellation patches minus one, max = 127
109 * # [7:11] = the number of output patch control points minus one, max = 31
110 * # [12:16] = the number of input patch control points minus one, max = 31
111 * # [17:22] = the number of LS outputs, up to 32
112 * # [23:28] = the number of HS per-vertex outputs, up to 32
113 * # [29:30] = tess_primitive_mode
114 * # [31] = whether TES reads tess factors
115 */
116 struct ac_arg tcs_offchip_layout;
117
118 /* GS */
119 struct ac_arg vgt_esgs_ring_itemsize;
120
121 /* For non-monolithic VS or TES on GFX9+. */
122 struct ac_arg next_stage_pc;
123
124 /* PS/TCS epilogs PC. */
125 struct ac_arg epilog_pc;
126
127 struct radv_userdata_locations user_sgprs_locs;
128 unsigned num_user_sgprs;
129
130 bool explicit_scratch_args;
131 bool remap_spi_ps_input;
132 bool load_grid_size_from_user_sgpr;
133 };
134
135 static inline struct radv_shader_args *
radv_shader_args_from_ac(struct ac_shader_args * args)136 radv_shader_args_from_ac(struct ac_shader_args *args)
137 {
138 return container_of(args, struct radv_shader_args, ac);
139 }
140
141 struct radv_graphics_state_key;
142 struct radv_shader_info;
143 struct radv_ps_epilog_key;
144 struct radv_device;
145
146 void radv_declare_shader_args(const struct radv_device *device, const struct radv_graphics_state_key *gfx_state,
147 const struct radv_shader_info *info, gl_shader_stage stage,
148 gl_shader_stage previous_stage, struct radv_shader_args *args);
149
150 void radv_declare_ps_epilog_args(const struct radv_device *device, const struct radv_ps_epilog_key *key,
151 struct radv_shader_args *args);
152
153 void radv_declare_rt_shader_args(enum amd_gfx_level gfx_level, struct radv_shader_args *args);
154
155 #endif /* RADV_SHADER_ARGS_H */
156