1 /* 2 * Copyright 2019 Valve Corporation 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a 5 * copy of this software and associated documentation files (the "Software"), 6 * to deal in the Software without restriction, including without limitation 7 * on the rights to use, copy, modify, merge, publish, distribute, sub 8 * license, and/or sell copies of the Software, and to permit persons to whom 9 * the Software is furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice (including the next 12 * paragraph) shall be included in all copies or substantial portions of the 13 * Software. 14 * 15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL 18 * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, 19 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR 20 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE 21 * USE OR OTHER DEALINGS IN THE SOFTWARE. 22 */ 23 24 #ifndef AC_SHADER_ARGS_H 25 #define AC_SHADER_ARGS_H 26 27 #include <stdbool.h> 28 #include <stdint.h> 29 30 /* Maximum dwords of inline push constants when the indirect path is still used */ 31 #define AC_MAX_INLINE_PUSH_CONSTS_WITH_INDIRECT 8 32 /* Maximum dwords of inline push constants when the indirect path is not used */ 33 #define AC_MAX_INLINE_PUSH_CONSTS 32 34 35 enum ac_arg_regfile 36 { 37 AC_ARG_SGPR, 38 AC_ARG_VGPR, 39 }; 40 41 enum ac_arg_type 42 { 43 AC_ARG_FLOAT, 44 AC_ARG_INT, 45 AC_ARG_CONST_PTR, /* Pointer to i8 array */ 46 AC_ARG_CONST_FLOAT_PTR, /* Pointer to f32 array */ 47 AC_ARG_CONST_PTR_PTR, /* Pointer to pointer to i8 array */ 48 AC_ARG_CONST_DESC_PTR, /* Pointer to v4i32 array */ 49 AC_ARG_CONST_IMAGE_PTR, /* Pointer to v8i32 array */ 50 }; 51 52 struct ac_arg { 53 uint16_t arg_index; 54 bool used; 55 }; 56 57 #define AC_MAX_ARGS 384 /* including all VS->TCS IO */ 58 59 struct ac_shader_args { 60 /* Info on how to declare arguments */ 61 struct { 62 enum ac_arg_type type; 63 enum ac_arg_regfile file; 64 uint8_t offset; 65 uint8_t size; 66 bool skip; 67 } args[AC_MAX_ARGS]; 68 69 uint16_t arg_count; 70 uint16_t num_sgprs_used; 71 uint16_t num_vgprs_used; 72 73 uint16_t return_count; 74 uint16_t num_sgprs_returned; 75 uint16_t num_vgprs_returned; 76 77 /* VS */ 78 struct ac_arg base_vertex; 79 struct ac_arg start_instance; 80 struct ac_arg draw_id; 81 struct ac_arg vertex_buffers; 82 struct ac_arg vertex_id; 83 struct ac_arg vs_rel_patch_id; 84 struct ac_arg vs_prim_id; 85 struct ac_arg instance_id; 86 87 /* Merged shaders */ 88 struct ac_arg tess_offchip_offset; 89 struct ac_arg merged_wave_info; 90 /* On gfx10: 91 * - bits 0..11: ordered_wave_id 92 * - bits 12..20: number of vertices in group 93 * - bits 22..30: number of primitives in group 94 */ 95 struct ac_arg gs_tg_info; 96 struct ac_arg scratch_offset; 97 98 /* TCS */ 99 struct ac_arg tcs_factor_offset; 100 struct ac_arg tcs_wave_id; /* gfx11+ */ 101 struct ac_arg tcs_patch_id; 102 struct ac_arg tcs_rel_ids; 103 104 /* TES */ 105 struct ac_arg tes_u; 106 struct ac_arg tes_v; 107 struct ac_arg tes_rel_patch_id; 108 struct ac_arg tes_patch_id; 109 110 /* GS */ 111 struct ac_arg es2gs_offset; /* separate legacy ES */ 112 struct ac_arg gs2vs_offset; /* legacy GS */ 113 struct ac_arg gs_wave_id; /* legacy GS */ 114 struct ac_arg gs_attr_offset; /* gfx11+: attribute ring offset in 512B increments */ 115 struct ac_arg gs_vtx_offset[6]; /* GFX6-8: [0-5], GFX9+: [0-2] packed */ 116 struct ac_arg gs_prim_id; 117 struct ac_arg gs_invocation_id; 118 119 /* Streamout */ 120 struct ac_arg streamout_config; 121 struct ac_arg streamout_write_index; 122 struct ac_arg streamout_offset[4]; 123 124 /* PS */ 125 struct ac_arg frag_pos[4]; 126 struct ac_arg front_face; 127 struct ac_arg ancillary; 128 struct ac_arg sample_coverage; 129 struct ac_arg prim_mask; 130 struct ac_arg persp_sample; 131 struct ac_arg persp_center; 132 struct ac_arg persp_centroid; 133 struct ac_arg pull_model; 134 struct ac_arg linear_sample; 135 struct ac_arg linear_center; 136 struct ac_arg linear_centroid; 137 138 /* CS */ 139 struct ac_arg local_invocation_ids; 140 struct ac_arg num_work_groups; 141 struct ac_arg workgroup_ids[3]; 142 struct ac_arg tg_size; 143 144 /* Mesh and task shaders */ 145 struct ac_arg task_ring_entry; /* Pointer into the draw and payload rings. */ 146 147 /* Vulkan only */ 148 struct ac_arg push_constants; 149 struct ac_arg inline_push_consts[AC_MAX_INLINE_PUSH_CONSTS]; 150 uint64_t inline_push_const_mask; 151 struct ac_arg view_index; 152 struct ac_arg sbt_descriptors; 153 struct ac_arg ray_launch_size_addr; 154 struct ac_arg force_vrs_rates; 155 }; 156 157 void ac_add_arg(struct ac_shader_args *info, enum ac_arg_regfile regfile, unsigned registers, 158 enum ac_arg_type type, struct ac_arg *arg); 159 void ac_add_return(struct ac_shader_args *info, enum ac_arg_regfile regfile); 160 161 #endif 162