radv/gfx10: Only set HW edge flags with gs & tess disabled.
[mesa.git] / src / amd / vulkan / radv_pipeline.c
index 6c0d0994cdbf946e010f54f5c2104d1d52d73ed6..5751440f3010353949eb46d850098494965723c0 100644 (file)
@@ -96,6 +96,30 @@ struct radv_gs_state {
        uint32_t lds_size;
 };
 
+struct radv_ngg_state {
+       uint16_t ngg_emit_size; /* in dwords */
+       uint32_t hw_max_esverts;
+       uint32_t max_gsprims;
+       uint32_t max_out_verts;
+       uint32_t prim_amp_factor;
+       uint32_t vgt_esgs_ring_itemsize;
+       bool max_vert_out_per_gs_instance;
+};
+
+bool radv_pipeline_has_ngg(const struct radv_pipeline *pipeline)
+{
+       struct radv_shader_variant *variant = NULL;
+       if (pipeline->shaders[MESA_SHADER_GEOMETRY])
+               variant = pipeline->shaders[MESA_SHADER_GEOMETRY];
+       else if (pipeline->shaders[MESA_SHADER_TESS_EVAL])
+               variant = pipeline->shaders[MESA_SHADER_TESS_EVAL];
+       else if (pipeline->shaders[MESA_SHADER_VERTEX])
+               variant = pipeline->shaders[MESA_SHADER_VERTEX];
+       else
+               return false;
+       return variant->info.is_ngg;
+}
+
 static void
 radv_pipeline_destroy(struct radv_device *device,
                       struct radv_pipeline *pipeline,
@@ -541,10 +565,13 @@ radv_pipeline_compute_spi_color_formats(struct radv_pipeline *pipeline,
                }
        }
 
-       blend->cb_shader_mask = ac_get_cb_shader_mask(col_format);
-
+       /* The output for dual source blending should have the same format as
+        * the first output.
+        */
        if (blend->mrt0_is_dual_src)
                col_format |= (col_format & 0xf) << 4;
+
+       blend->cb_shader_mask = ac_get_cb_shader_mask(col_format);
        blend->spi_shader_col_format = col_format;
 }
 
@@ -1580,6 +1607,203 @@ calculate_gs_info(const VkGraphicsPipelineCreateInfo *pCreateInfo,
        return gs;
 }
 
+static void clamp_gsprims_to_esverts(unsigned *max_gsprims, unsigned max_esverts,
+                                    unsigned min_verts_per_prim, bool use_adjacency)
+{
+       unsigned max_reuse = max_esverts - min_verts_per_prim;
+       if (use_adjacency)
+               max_reuse /= 2;
+       *max_gsprims = MIN2(*max_gsprims, 1 + max_reuse);
+}
+
+static struct radv_ngg_state
+calculate_ngg_info(const VkGraphicsPipelineCreateInfo *pCreateInfo,
+                  struct radv_pipeline *pipeline)
+{
+       struct radv_ngg_state ngg = {0};
+       struct radv_shader_variant_info *gs_info = &pipeline->shaders[MESA_SHADER_GEOMETRY]->info;
+       struct radv_es_output_info *es_info =
+               radv_pipeline_has_tess(pipeline) ? &gs_info->tes.es_info : &gs_info->vs.es_info;
+       unsigned gs_type = MESA_SHADER_VERTEX;
+       unsigned max_verts_per_prim = 3; // triangles
+       unsigned min_verts_per_prim =
+               gs_type == MESA_SHADER_GEOMETRY ? max_verts_per_prim : 1;
+       unsigned gs_num_invocations = 1;//MAX2(gs_info->gs.invocations, 1);
+       bool uses_adjacency;
+       switch(pCreateInfo->pInputAssemblyState->topology) {
+       case VK_PRIMITIVE_TOPOLOGY_LINE_LIST_WITH_ADJACENCY:
+       case VK_PRIMITIVE_TOPOLOGY_LINE_STRIP_WITH_ADJACENCY:
+       case VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST_WITH_ADJACENCY:
+       case VK_PRIMITIVE_TOPOLOGY_TRIANGLE_STRIP_WITH_ADJACENCY:
+               uses_adjacency = true;
+               break;
+       default:
+               uses_adjacency = false;
+               break;
+       }
+
+       /* All these are in dwords: */
+       /* We can't allow using the whole LDS, because GS waves compete with
+        * other shader stages for LDS space.
+        *
+        * Streamout can increase the ESGS buffer size later on, so be more
+        * conservative with streamout and use 4K dwords. This may be suboptimal.
+        *
+        * Otherwise, use the limit of 7K dwords. The reason is that we need
+        * to leave some headroom for the max_esverts increase at the end.
+        *
+        * TODO: We should really take the shader's internal LDS use into
+        *       account. The linker will fail if the size is greater than
+        *       8K dwords.
+        */
+       const unsigned max_lds_size = (0 /*gs_info->info.so.num_outputs*/ ? 4 : 7) * 1024 - 128;
+       const unsigned target_lds_size = max_lds_size;
+       unsigned esvert_lds_size = 0;
+       unsigned gsprim_lds_size = 0;
+
+       /* All these are per subgroup: */
+       bool max_vert_out_per_gs_instance = false;
+       unsigned max_esverts_base = 256;
+       unsigned max_gsprims_base = 128; /* default prim group size clamp */
+
+       /* Hardware has the following non-natural restrictions on the value
+        * of GE_CNTL.VERT_GRP_SIZE based on based on the primitive type of
+        * the draw:
+        *  - at most 252 for any line input primitive type
+        *  - at most 251 for any quad input primitive type
+        *  - at most 251 for triangle strips with adjacency (this happens to
+        *    be the natural limit for triangle *lists* with adjacency)
+        */
+       max_esverts_base = MIN2(max_esverts_base, 251 + max_verts_per_prim - 1);
+
+       if (gs_type == MESA_SHADER_GEOMETRY) {
+               unsigned max_out_verts_per_gsprim =
+                       gs_info->gs.vertices_out * gs_num_invocations;
+
+               if (max_out_verts_per_gsprim <= 256) {
+                       if (max_out_verts_per_gsprim) {
+                               max_gsprims_base = MIN2(max_gsprims_base,
+                                                       256 / max_out_verts_per_gsprim);
+                       }
+               } else {
+                       /* Use special multi-cycling mode in which each GS
+                        * instance gets its own subgroup. Does not work with
+                        * tessellation. */
+                       max_vert_out_per_gs_instance = true;
+                       max_gsprims_base = 1;
+                       max_out_verts_per_gsprim = gs_info->gs.vertices_out;
+               }
+
+               esvert_lds_size = es_info->esgs_itemsize / 4;
+               gsprim_lds_size = (gs_info->gs.gsvs_vertex_size / 4 + 1) * max_out_verts_per_gsprim;
+       } else {
+               /* TODO: This needs to be adjusted once LDS use for compaction
+                * after culling is implemented. */
+               /*
+               if (es_info->info.so.num_outputs)
+                       esvert_lds_size = 4 * es_info->info.so.num_outputs + 1;
+               */
+       }
+
+       unsigned max_gsprims = max_gsprims_base;
+       unsigned max_esverts = max_esverts_base;
+
+       if (esvert_lds_size)
+               max_esverts = MIN2(max_esverts, target_lds_size / esvert_lds_size);
+       if (gsprim_lds_size)
+               max_gsprims = MIN2(max_gsprims, target_lds_size / gsprim_lds_size);
+
+       max_esverts = MIN2(max_esverts, max_gsprims * max_verts_per_prim);
+       clamp_gsprims_to_esverts(&max_gsprims, max_esverts, min_verts_per_prim, uses_adjacency);
+       assert(max_esverts >= max_verts_per_prim && max_gsprims >= 1);
+
+       if (esvert_lds_size || gsprim_lds_size) {
+               /* Now that we have a rough proportionality between esverts
+                * and gsprims based on the primitive type, scale both of them
+                * down simultaneously based on required LDS space.
+                *
+                * We could be smarter about this if we knew how much vertex
+                * reuse to expect.
+                */
+               unsigned lds_total = max_esverts * esvert_lds_size +
+                                    max_gsprims * gsprim_lds_size;
+               if (lds_total > target_lds_size) {
+                       max_esverts = max_esverts * target_lds_size / lds_total;
+                       max_gsprims = max_gsprims * target_lds_size / lds_total;
+
+                       max_esverts = MIN2(max_esverts, max_gsprims * max_verts_per_prim);
+                       clamp_gsprims_to_esverts(&max_gsprims, max_esverts,
+                                                min_verts_per_prim, uses_adjacency);
+                       assert(max_esverts >= max_verts_per_prim && max_gsprims >= 1);
+               }
+       }
+
+       /* Round up towards full wave sizes for better ALU utilization. */
+       if (!max_vert_out_per_gs_instance) {
+               const unsigned wavesize = 64;
+               unsigned orig_max_esverts;
+               unsigned orig_max_gsprims;
+               do {
+                       orig_max_esverts = max_esverts;
+                       orig_max_gsprims = max_gsprims;
+
+                       max_esverts = align(max_esverts, wavesize);
+                       max_esverts = MIN2(max_esverts, max_esverts_base);
+                       if (esvert_lds_size)
+                               max_esverts = MIN2(max_esverts,
+                                                  (max_lds_size - max_gsprims * gsprim_lds_size) /
+                                                  esvert_lds_size);
+                       max_esverts = MIN2(max_esverts, max_gsprims * max_verts_per_prim);
+
+                       max_gsprims = align(max_gsprims, wavesize);
+                       max_gsprims = MIN2(max_gsprims, max_gsprims_base);
+                       if (gsprim_lds_size)
+                               max_gsprims = MIN2(max_gsprims,
+                                                  (max_lds_size - max_esverts * esvert_lds_size) /
+                                                  gsprim_lds_size);
+                       clamp_gsprims_to_esverts(&max_gsprims, max_esverts,
+                                                min_verts_per_prim, uses_adjacency);
+                       assert(max_esverts >= max_verts_per_prim && max_gsprims >= 1);
+               } while (orig_max_esverts != max_esverts || orig_max_gsprims != max_gsprims);
+       }
+
+       /* Hardware restriction: minimum value of max_esverts */
+       max_esverts = MAX2(max_esverts, 23 + max_verts_per_prim);
+
+       unsigned max_out_vertices =
+               max_vert_out_per_gs_instance ? gs_info->gs.vertices_out :
+               gs_type == MESA_SHADER_GEOMETRY ?
+               max_gsprims * gs_num_invocations * gs_info->gs.vertices_out :
+               max_esverts;
+       assert(max_out_vertices <= 256);
+
+       unsigned prim_amp_factor = 1;
+       if (gs_type == MESA_SHADER_GEOMETRY) {
+               /* Number of output primitives per GS input primitive after
+                * GS instancing. */
+               prim_amp_factor = gs_info->gs.vertices_out;
+       }
+
+       /* The GE only checks against the maximum number of ES verts after
+        * allocating a full GS primitive. So we need to ensure that whenever
+        * this check passes, there is enough space for a full primitive without
+        * vertex reuse.
+        */
+       ngg.hw_max_esverts = max_esverts - max_verts_per_prim + 1;
+       ngg.max_gsprims = max_gsprims;
+       ngg.max_out_verts = max_out_vertices;
+       ngg.prim_amp_factor = prim_amp_factor;
+       ngg.max_vert_out_per_gs_instance = max_vert_out_per_gs_instance;
+       ngg.ngg_emit_size = max_gsprims * gsprim_lds_size;
+       ngg.vgt_esgs_ring_itemsize = 1;
+
+       pipeline->graphics.esgs_ring_size = 4 * max_esverts * esvert_lds_size;
+
+       assert(ngg.hw_max_esverts >= 24); /* HW limitation */
+
+       return ngg;
+}
+
 static void
 calculate_gs_ring_sizes(struct radv_pipeline *pipeline, const struct radv_gs_state *gs)
 {
@@ -1997,7 +2221,8 @@ radv_generate_graphics_pipeline_key(struct radv_pipeline *pipeline,
 }
 
 static void
-radv_fill_shader_keys(struct radv_shader_variant_key *keys,
+radv_fill_shader_keys(struct radv_device *device,
+                     struct radv_shader_variant_key *keys,
                       const struct radv_pipeline_key *key,
                       nir_shader **nir)
 {
@@ -2013,7 +2238,7 @@ radv_fill_shader_keys(struct radv_shader_variant_key *keys,
        }
 
        if (nir[MESA_SHADER_TESS_CTRL]) {
-               keys[MESA_SHADER_VERTEX].vs.as_ls = true;
+               keys[MESA_SHADER_VERTEX].vs.out.as_ls = true;
                keys[MESA_SHADER_TESS_CTRL].tcs.num_inputs = 0;
                keys[MESA_SHADER_TESS_CTRL].tcs.input_vertices = key->tess_input_vertices;
                keys[MESA_SHADER_TESS_CTRL].tcs.primitive_mode = nir[MESA_SHADER_TESS_EVAL]->info.tess.primitive_mode;
@@ -2023,9 +2248,13 @@ radv_fill_shader_keys(struct radv_shader_variant_key *keys,
 
        if (nir[MESA_SHADER_GEOMETRY]) {
                if (nir[MESA_SHADER_TESS_CTRL])
-                       keys[MESA_SHADER_TESS_EVAL].tes.as_es = true;
+                       keys[MESA_SHADER_TESS_EVAL].tes.out.as_es = true;
                else
-                       keys[MESA_SHADER_VERTEX].vs.as_es = true;
+                       keys[MESA_SHADER_VERTEX].vs.out.as_es = true;
+       }
+
+       if (device->physical_device->rad_info.chip_class >= GFX10) {
+               keys[MESA_SHADER_VERTEX].vs.out.as_ngg = true;
        }
 
        for(int i = 0; i < MESA_SHADER_STAGES; ++i)
@@ -2128,9 +2357,8 @@ void radv_create_shaders(struct radv_pipeline *pipeline,
        struct radv_shader_module fs_m = {0};
        struct radv_shader_module *modules[MESA_SHADER_STAGES] = { 0, };
        nir_shader *nir[MESA_SHADER_STAGES] = {0};
-       void *codes[MESA_SHADER_STAGES] = {0};
-       unsigned code_sizes[MESA_SHADER_STAGES] = {0};
-       struct radv_shader_variant_key keys[MESA_SHADER_STAGES] = {{{{0}}}};
+       struct radv_shader_binary *binaries[MESA_SHADER_STAGES] = {NULL};
+       struct radv_shader_variant_key keys[MESA_SHADER_STAGES] = {{{{{0}}}}};
        unsigned char hash[20], gs_copy_hash[20];
 
        radv_start_feedback(pipeline_feedback);
@@ -2219,29 +2447,33 @@ void radv_create_shaders(struct radv_pipeline *pipeline,
                        nir_print_shader(nir[i], stderr);
        }
 
-       radv_fill_shader_keys(keys, key, nir);
+       radv_fill_shader_keys(device, keys, key, nir);
 
        if (nir[MESA_SHADER_FRAGMENT]) {
                if (!pipeline->shaders[MESA_SHADER_FRAGMENT]) {
                        radv_start_feedback(stage_feedbacks[MESA_SHADER_FRAGMENT]);
 
                        pipeline->shaders[MESA_SHADER_FRAGMENT] =
-                              radv_shader_variant_create(device, modules[MESA_SHADER_FRAGMENT], &nir[MESA_SHADER_FRAGMENT], 1,
+                              radv_shader_variant_compile(device, modules[MESA_SHADER_FRAGMENT], &nir[MESA_SHADER_FRAGMENT], 1,
                                                          pipeline->layout, keys + MESA_SHADER_FRAGMENT,
-                                                         &codes[MESA_SHADER_FRAGMENT], &code_sizes[MESA_SHADER_FRAGMENT]);
+                                                         &binaries[MESA_SHADER_FRAGMENT]);
 
                        radv_stop_feedback(stage_feedbacks[MESA_SHADER_FRAGMENT], false);
                }
 
                /* TODO: These are no longer used as keys we should refactor this */
-               keys[MESA_SHADER_VERTEX].vs.export_prim_id =
+               keys[MESA_SHADER_VERTEX].vs.out.export_prim_id =
                        pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.prim_id_input;
-               keys[MESA_SHADER_VERTEX].vs.export_layer_id =
+               keys[MESA_SHADER_VERTEX].vs.out.export_layer_id =
                        pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.layer_input;
-               keys[MESA_SHADER_TESS_EVAL].tes.export_prim_id =
+               keys[MESA_SHADER_VERTEX].vs.out.export_clip_dists =
+                       !!pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.num_input_clips_culls;
+               keys[MESA_SHADER_TESS_EVAL].tes.out.export_prim_id =
                        pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.prim_id_input;
-               keys[MESA_SHADER_TESS_EVAL].tes.export_layer_id =
+               keys[MESA_SHADER_TESS_EVAL].tes.out.export_layer_id =
                        pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.layer_input;
+               keys[MESA_SHADER_TESS_EVAL].tes.out.export_clip_dists =
+                       !!pipeline->shaders[MESA_SHADER_FRAGMENT]->info.info.ps.num_input_clips_culls;
        }
 
        if (device->physical_device->rad_info.chip_class >= GFX9 && modules[MESA_SHADER_TESS_CTRL]) {
@@ -2252,10 +2484,9 @@ void radv_create_shaders(struct radv_pipeline *pipeline,
 
                        radv_start_feedback(stage_feedbacks[MESA_SHADER_TESS_CTRL]);
 
-                       pipeline->shaders[MESA_SHADER_TESS_CTRL] = radv_shader_variant_create(device, modules[MESA_SHADER_TESS_CTRL], combined_nir, 2,
+                       pipeline->shaders[MESA_SHADER_TESS_CTRL] = radv_shader_variant_compile(device, modules[MESA_SHADER_TESS_CTRL], combined_nir, 2,
                                                                                              pipeline->layout,
-                                                                                             &key, &codes[MESA_SHADER_TESS_CTRL],
-                                                                                             &code_sizes[MESA_SHADER_TESS_CTRL]);
+                                                                                             &key, &binaries[MESA_SHADER_TESS_CTRL]);
 
                        radv_stop_feedback(stage_feedbacks[MESA_SHADER_TESS_CTRL], false);
                }
@@ -2271,10 +2502,9 @@ void radv_create_shaders(struct radv_pipeline *pipeline,
 
                        radv_start_feedback(stage_feedbacks[MESA_SHADER_GEOMETRY]);
 
-                       pipeline->shaders[MESA_SHADER_GEOMETRY] = radv_shader_variant_create(device, modules[MESA_SHADER_GEOMETRY], combined_nir, 2,
+                       pipeline->shaders[MESA_SHADER_GEOMETRY] = radv_shader_variant_compile(device, modules[MESA_SHADER_GEOMETRY], combined_nir, 2,
                                                                                             pipeline->layout,
-                                                                                            &keys[pre_stage] , &codes[MESA_SHADER_GEOMETRY],
-                                                                                    &code_sizes[MESA_SHADER_GEOMETRY]);
+                                                                                            &keys[pre_stage] , &binaries[MESA_SHADER_GEOMETRY]);
 
                        radv_stop_feedback(stage_feedbacks[MESA_SHADER_GEOMETRY], false);
                }
@@ -2293,48 +2523,42 @@ void radv_create_shaders(struct radv_pipeline *pipeline,
 
                        radv_start_feedback(stage_feedbacks[i]);
 
-                       pipeline->shaders[i] = radv_shader_variant_create(device, modules[i], &nir[i], 1,
+                       pipeline->shaders[i] = radv_shader_variant_compile(device, modules[i], &nir[i], 1,
                                                                          pipeline->layout,
-                                                                         keys + i, &codes[i],
-                                                                         &code_sizes[i]);
+                                                                         keys + i, &binaries[i]);
 
                        radv_stop_feedback(stage_feedbacks[i], false);
                }
        }
 
        if(modules[MESA_SHADER_GEOMETRY]) {
-               void *gs_copy_code = NULL;
-               unsigned gs_copy_code_size = 0;
+               struct radv_shader_binary *gs_copy_binary = NULL;
                if (!pipeline->gs_copy_shader) {
                        pipeline->gs_copy_shader = radv_create_gs_copy_shader(
-                                       device, nir[MESA_SHADER_GEOMETRY], &gs_copy_code,
-                                       &gs_copy_code_size,
+                                       device, nir[MESA_SHADER_GEOMETRY], &gs_copy_binary,
                                        keys[MESA_SHADER_GEOMETRY].has_multiview_view_index);
                }
 
                if (pipeline->gs_copy_shader) {
-                       void *code[MESA_SHADER_STAGES] = {0};
-                       unsigned code_size[MESA_SHADER_STAGES] = {0};
+                       struct radv_shader_binary *binaries[MESA_SHADER_STAGES] = {NULL};
                        struct radv_shader_variant *variants[MESA_SHADER_STAGES] = {0};
 
-                       code[MESA_SHADER_GEOMETRY] = gs_copy_code;
-                       code_size[MESA_SHADER_GEOMETRY] = gs_copy_code_size;
+                       binaries[MESA_SHADER_GEOMETRY] = gs_copy_binary;
                        variants[MESA_SHADER_GEOMETRY] = pipeline->gs_copy_shader;
 
                        radv_pipeline_cache_insert_shaders(device, cache,
                                                           gs_copy_hash,
                                                           variants,
-                                                          (const void**)code,
-                                                          code_size);
+                                                          binaries);
                }
-               free(gs_copy_code);
+               free(gs_copy_binary);
        }
 
        radv_pipeline_cache_insert_shaders(device, cache, hash, pipeline->shaders,
-                                          (const void**)codes, code_sizes);
+                                          binaries);
 
        for (int i = 0; i < MESA_SHADER_STAGES; ++i) {
-               free(codes[i]);
+               free(binaries[i]);
                if (nir[i]) {
                        if (!pipeline->device->keep_shader_info)
                                ralloc_free(nir[i]);
@@ -2358,36 +2582,50 @@ radv_pipeline_stage_to_user_data_0(struct radv_pipeline *pipeline,
 {
        bool has_gs = radv_pipeline_has_gs(pipeline);
        bool has_tess = radv_pipeline_has_tess(pipeline);
+       bool has_ngg = radv_pipeline_has_ngg(pipeline);
+
        switch (stage) {
        case MESA_SHADER_FRAGMENT:
                return R_00B030_SPI_SHADER_USER_DATA_PS_0;
        case MESA_SHADER_VERTEX:
-               if (chip_class >= GFX9) {
-                       return has_tess ? R_00B430_SPI_SHADER_USER_DATA_LS_0 :
-                              has_gs ? R_00B330_SPI_SHADER_USER_DATA_ES_0 :
-                              R_00B130_SPI_SHADER_USER_DATA_VS_0;
+               if (has_tess) {
+                       if (chip_class >= GFX10) {
+                               return R_00B430_SPI_SHADER_USER_DATA_HS_0;
+                       } else if (chip_class == GFX9) {
+                               return R_00B430_SPI_SHADER_USER_DATA_LS_0;
+                       } else {
+                               return R_00B530_SPI_SHADER_USER_DATA_LS_0;
+                       }
+
                }
-               if (has_tess)
-                       return R_00B530_SPI_SHADER_USER_DATA_LS_0;
-               else
-                       return has_gs ? R_00B330_SPI_SHADER_USER_DATA_ES_0 : R_00B130_SPI_SHADER_USER_DATA_VS_0;
+
+               if (has_gs) {
+                       if (chip_class >= GFX10) {
+                               return R_00B230_SPI_SHADER_USER_DATA_GS_0;
+                       } else {
+                               return R_00B330_SPI_SHADER_USER_DATA_ES_0;
+                       }
+               }
+
+               if (has_ngg)
+                       return R_00B230_SPI_SHADER_USER_DATA_GS_0;
+
+               return R_00B130_SPI_SHADER_USER_DATA_VS_0;
        case MESA_SHADER_GEOMETRY:
-               return chip_class >= GFX9 ? R_00B330_SPI_SHADER_USER_DATA_ES_0 :
+               return chip_class == GFX9 ? R_00B330_SPI_SHADER_USER_DATA_ES_0 :
                                            R_00B230_SPI_SHADER_USER_DATA_GS_0;
        case MESA_SHADER_COMPUTE:
                return R_00B900_COMPUTE_USER_DATA_0;
        case MESA_SHADER_TESS_CTRL:
-               return chip_class >= GFX9 ? R_00B430_SPI_SHADER_USER_DATA_LS_0 :
+               return chip_class == GFX9 ? R_00B430_SPI_SHADER_USER_DATA_LS_0 :
                                            R_00B430_SPI_SHADER_USER_DATA_HS_0;
        case MESA_SHADER_TESS_EVAL:
-               if (chip_class >= GFX9) {
-                       return has_gs ? R_00B330_SPI_SHADER_USER_DATA_ES_0 :
-                              R_00B130_SPI_SHADER_USER_DATA_VS_0;
-               }
-               if (has_gs)
-                       return R_00B330_SPI_SHADER_USER_DATA_ES_0;
-               else
+               if (has_gs) {
+                       return chip_class >= GFX10 ? R_00B230_SPI_SHADER_USER_DATA_GS_0 :
+                                                    R_00B330_SPI_SHADER_USER_DATA_ES_0;
+               } else {
                        return R_00B130_SPI_SHADER_USER_DATA_VS_0;
+               }
        default:
                unreachable("unknown shader");
        }
@@ -2684,29 +2922,29 @@ radv_pipeline_generate_binning_state(struct radeon_cmdbuf *ctx_cs,
 
        VkExtent2D bin_size = radv_compute_bin_size(pipeline, pCreateInfo);
 
-       unsigned context_states_per_bin; /* allowed range: [1, 6] */
-       unsigned persistent_states_per_bin; /* allowed range: [1, 32] */
-       unsigned fpovs_per_batch; /* allowed range: [0, 255], 0 = unlimited */
-
-       switch (pipeline->device->physical_device->rad_info.family) {
-       case CHIP_VEGA10:
-       case CHIP_VEGA12:
-       case CHIP_VEGA20:
-               context_states_per_bin = 1;
-               persistent_states_per_bin = 1;
-               fpovs_per_batch = 63;
-               break;
-       case CHIP_RAVEN:
-       case CHIP_RAVEN2:
-               context_states_per_bin = 6;
-               persistent_states_per_bin = 32;
-               fpovs_per_batch = 63;
-               break;
-       default:
-               unreachable("unhandled family while determining binning state.");
-       }
-
        if (pipeline->device->pbb_allowed && bin_size.width && bin_size.height) {
+               unsigned context_states_per_bin; /* allowed range: [1, 6] */
+               unsigned persistent_states_per_bin; /* allowed range: [1, 32] */
+               unsigned fpovs_per_batch; /* allowed range: [0, 255], 0 = unlimited */
+
+               switch (pipeline->device->physical_device->rad_info.family) {
+               case CHIP_VEGA10:
+               case CHIP_VEGA12:
+               case CHIP_VEGA20:
+                       context_states_per_bin = 1;
+                       persistent_states_per_bin = 1;
+                       fpovs_per_batch = 63;
+                       break;
+               case CHIP_RAVEN:
+               case CHIP_RAVEN2:
+                       context_states_per_bin = 6;
+                       persistent_states_per_bin = 32;
+                       fpovs_per_batch = 63;
+                       break;
+               default:
+                       unreachable("unhandled family while determining binning state.");
+               }
+
                pa_sc_binner_cntl_0 =
                        S_028C44_BINNING_MODE(V_028C44_BINNING_ALLOWED) |
                        S_028C44_BIN_SIZE_X(bin_size.width == 16) |
@@ -2722,8 +2960,14 @@ radv_pipeline_generate_binning_state(struct radeon_cmdbuf *ctx_cs,
 
        radeon_set_context_reg(ctx_cs, R_028C44_PA_SC_BINNER_CNTL_0,
                               pa_sc_binner_cntl_0);
-       radeon_set_context_reg(ctx_cs, R_028060_DB_DFSM_CONTROL,
-                              db_dfsm_control);
+
+       if (pipeline->device->physical_device->rad_info.chip_class >= GFX10) {
+               radeon_set_context_reg(ctx_cs, R_028038_DB_DFSM_CONTROL,
+                                      db_dfsm_control);
+       } else {
+               radeon_set_context_reg(ctx_cs, R_028060_DB_DFSM_CONTROL,
+                                      db_dfsm_control);
+       }
 }
 
 
@@ -2736,7 +2980,6 @@ radv_pipeline_generate_depth_stencil_state(struct radeon_cmdbuf *ctx_cs,
        const VkPipelineDepthStencilStateCreateInfo *vkds = pCreateInfo->pDepthStencilState;
        RADV_FROM_HANDLE(radv_render_pass, pass, pCreateInfo->renderPass);
        struct radv_subpass *subpass = pass->subpasses + pCreateInfo->subpass;
-       struct radv_shader_variant *ps = pipeline->shaders[MESA_SHADER_FRAGMENT];
        struct radv_render_pass_attachment *attachment = NULL;
        uint32_t db_depth_control = 0, db_stencil_control = 0;
        uint32_t db_render_control = 0, db_render_override2 = 0;
@@ -2785,8 +3028,7 @@ radv_pipeline_generate_depth_stencil_state(struct radeon_cmdbuf *ctx_cs,
        db_render_override |= S_02800C_FORCE_HIS_ENABLE0(V_02800C_FORCE_DISABLE) |
                              S_02800C_FORCE_HIS_ENABLE1(V_02800C_FORCE_DISABLE);
 
-       if (!pCreateInfo->pRasterizationState->depthClampEnable &&
-           ps->info.info.ps.writes_z) {
+       if (!pCreateInfo->pRasterizationState->depthClampEnable) {
                /* From VK_EXT_depth_range_unrestricted spec:
                 *
                 * "The behavior described in Primitive Clipping still applies.
@@ -2957,8 +3199,7 @@ radv_pipeline_generate_vgt_gs_mode(struct radeon_cmdbuf *ctx_cs,
                                    struct radv_pipeline *pipeline)
 {
        const struct radv_vs_output_info *outinfo = get_vs_output_info(pipeline);
-
-       uint32_t vgt_primitiveid_en = false;
+       unsigned vgt_primitiveid_en = 0;
        uint32_t vgt_gs_mode = 0;
 
        if (radv_pipeline_has_gs(pipeline)) {
@@ -2967,9 +3208,17 @@ radv_pipeline_generate_vgt_gs_mode(struct radeon_cmdbuf *ctx_cs,
 
                vgt_gs_mode = ac_vgt_gs_mode(gs->info.gs.vertices_out,
                                             pipeline->device->physical_device->rad_info.chip_class);
+       } else if (radv_pipeline_has_ngg(pipeline)) {
+               const struct radv_shader_variant *vs =
+                       pipeline->shaders[MESA_SHADER_VERTEX];
+               bool enable_prim_id =
+                       outinfo->export_prim_id || vs->info.info.uses_prim_id;
+
+               vgt_primitiveid_en |= S_028A84_PRIMITIVEID_EN(enable_prim_id) |
+                                     S_028A84_NGG_DISABLE_PROVOK_REUSE(enable_prim_id);
        } else if (outinfo->export_prim_id) {
                vgt_gs_mode = S_028A40_MODE(V_028A40_GS_SCENARIO_A);
-               vgt_primitiveid_en = true;
+               vgt_primitiveid_en |= S_028A84_PRIMITIVEID_EN(1);
        }
 
        radeon_set_context_reg(ctx_cs, R_028A84_VGT_PRIMITIVEID_EN, vgt_primitiveid_en);
@@ -2987,8 +3236,8 @@ radv_pipeline_generate_hw_vs(struct radeon_cmdbuf *ctx_cs,
        radeon_set_sh_reg_seq(cs, R_00B120_SPI_SHADER_PGM_LO_VS, 4);
        radeon_emit(cs, va >> 8);
        radeon_emit(cs, S_00B124_MEM_BASE(va >> 40));
-       radeon_emit(cs, shader->rsrc1);
-       radeon_emit(cs, shader->rsrc2);
+       radeon_emit(cs, shader->config.rsrc1);
+       radeon_emit(cs, shader->config.rsrc2);
 
        const struct radv_vs_output_info *outinfo = get_vs_output_info(pipeline);
        unsigned clip_dist_mask, cull_dist_mask, total_mask;
@@ -3046,8 +3295,8 @@ radv_pipeline_generate_hw_es(struct radeon_cmdbuf *cs,
        radeon_set_sh_reg_seq(cs, R_00B320_SPI_SHADER_PGM_LO_ES, 4);
        radeon_emit(cs, va >> 8);
        radeon_emit(cs, S_00B324_MEM_BASE(va >> 40));
-       radeon_emit(cs, shader->rsrc1);
-       radeon_emit(cs, shader->rsrc2);
+       radeon_emit(cs, shader->config.rsrc1);
+       radeon_emit(cs, shader->config.rsrc2);
 }
 
 static void
@@ -3057,7 +3306,7 @@ radv_pipeline_generate_hw_ls(struct radeon_cmdbuf *cs,
                             const struct radv_tessellation_state *tess)
 {
        uint64_t va = radv_buffer_get_va(shader->bo) + shader->bo_offset;
-       uint32_t rsrc2 = shader->rsrc2;
+       uint32_t rsrc2 = shader->config.rsrc2;
 
        radeon_set_sh_reg_seq(cs, R_00B520_SPI_SHADER_PGM_LO_LS, 2);
        radeon_emit(cs, va >> 8);
@@ -3069,10 +3318,110 @@ radv_pipeline_generate_hw_ls(struct radeon_cmdbuf *cs,
                radeon_set_sh_reg(cs, R_00B52C_SPI_SHADER_PGM_RSRC2_LS, rsrc2);
 
        radeon_set_sh_reg_seq(cs, R_00B528_SPI_SHADER_PGM_RSRC1_LS, 2);
-       radeon_emit(cs, shader->rsrc1);
+       radeon_emit(cs, shader->config.rsrc1);
        radeon_emit(cs, rsrc2);
 }
 
+static void
+radv_pipeline_generate_hw_ngg(struct radeon_cmdbuf *ctx_cs,
+                             struct radeon_cmdbuf *cs,
+                             struct radv_pipeline *pipeline,
+                             struct radv_shader_variant *shader,
+                             const struct radv_ngg_state *ngg_state)
+{
+       uint64_t va = radv_buffer_get_va(shader->bo) + shader->bo_offset;
+
+       radeon_set_sh_reg_seq(cs, R_00B320_SPI_SHADER_PGM_LO_ES, 2);
+       radeon_emit(cs, va >> 8);
+       radeon_emit(cs, va >> 40);
+       radeon_set_sh_reg_seq(cs, R_00B228_SPI_SHADER_PGM_RSRC1_GS, 2);
+       radeon_emit(cs, shader->config.rsrc1);
+       radeon_emit(cs, shader->config.rsrc2);
+
+       const struct radv_vs_output_info *outinfo = get_vs_output_info(pipeline);
+       unsigned clip_dist_mask, cull_dist_mask, total_mask;
+       clip_dist_mask = outinfo->clip_dist_mask;
+       cull_dist_mask = outinfo->cull_dist_mask;
+       total_mask = clip_dist_mask | cull_dist_mask;
+       bool misc_vec_ena = outinfo->writes_pointsize ||
+               outinfo->writes_layer ||
+               outinfo->writes_viewport_index;
+       bool break_wave_at_eoi = false;
+
+       radeon_set_context_reg(ctx_cs, R_0286C4_SPI_VS_OUT_CONFIG,
+                              S_0286C4_VS_EXPORT_COUNT(MAX2(1, outinfo->param_exports) - 1));
+       radeon_set_context_reg(ctx_cs, R_028708_SPI_SHADER_IDX_FORMAT,
+                              S_028708_IDX0_EXPORT_FORMAT(V_028708_SPI_SHADER_1COMP));
+       radeon_set_context_reg(ctx_cs, R_02870C_SPI_SHADER_POS_FORMAT,
+                              S_02870C_POS0_EXPORT_FORMAT(V_02870C_SPI_SHADER_4COMP) |
+                              S_02870C_POS1_EXPORT_FORMAT(outinfo->pos_exports > 1 ?
+                                                          V_02870C_SPI_SHADER_4COMP :
+                                                          V_02870C_SPI_SHADER_NONE) |
+                              S_02870C_POS2_EXPORT_FORMAT(outinfo->pos_exports > 2 ?
+                                                          V_02870C_SPI_SHADER_4COMP :
+                                                          V_02870C_SPI_SHADER_NONE) |
+                              S_02870C_POS3_EXPORT_FORMAT(outinfo->pos_exports > 3 ?
+                                                          V_02870C_SPI_SHADER_4COMP :
+                                                          V_02870C_SPI_SHADER_NONE));
+
+       radeon_set_context_reg(ctx_cs, R_028818_PA_CL_VTE_CNTL,
+                              S_028818_VTX_W0_FMT(1) |
+                              S_028818_VPORT_X_SCALE_ENA(1) | S_028818_VPORT_X_OFFSET_ENA(1) |
+                              S_028818_VPORT_Y_SCALE_ENA(1) | S_028818_VPORT_Y_OFFSET_ENA(1) |
+                              S_028818_VPORT_Z_SCALE_ENA(1) | S_028818_VPORT_Z_OFFSET_ENA(1));
+       radeon_set_context_reg(ctx_cs, R_02881C_PA_CL_VS_OUT_CNTL,
+                              S_02881C_USE_VTX_POINT_SIZE(outinfo->writes_pointsize) |
+                              S_02881C_USE_VTX_RENDER_TARGET_INDX(outinfo->writes_layer) |
+                              S_02881C_USE_VTX_VIEWPORT_INDX(outinfo->writes_viewport_index) |
+                              S_02881C_VS_OUT_MISC_VEC_ENA(misc_vec_ena) |
+                              S_02881C_VS_OUT_MISC_SIDE_BUS_ENA(misc_vec_ena) |
+                              S_02881C_VS_OUT_CCDIST0_VEC_ENA((total_mask & 0x0f) != 0) |
+                              S_02881C_VS_OUT_CCDIST1_VEC_ENA((total_mask & 0xf0) != 0) |
+                              cull_dist_mask << 8 |
+                              clip_dist_mask);
+
+       /* TODO: Correctly set REUSE_OFF */
+       radeon_set_context_reg(ctx_cs, R_028AB4_VGT_REUSE_OFF,
+                              S_028AB4_REUSE_OFF(0));
+       radeon_set_context_reg(ctx_cs, R_028AAC_VGT_ESGS_RING_ITEMSIZE,
+                              ngg_state->vgt_esgs_ring_itemsize);
+
+       /* NGG specific registers. */
+       struct radv_shader_variant *gs = pipeline->shaders[MESA_SHADER_GEOMETRY];
+       uint32_t gs_num_invocations = gs ? gs->info.gs.invocations : 1;
+
+       radeon_set_context_reg(ctx_cs, R_028A44_VGT_GS_ONCHIP_CNTL,
+                              S_028A44_ES_VERTS_PER_SUBGRP(ngg_state->hw_max_esverts) |
+                              S_028A44_GS_PRIMS_PER_SUBGRP(ngg_state->max_gsprims) |
+                              S_028A44_GS_INST_PRIMS_IN_SUBGRP(ngg_state->max_gsprims * gs_num_invocations));
+       radeon_set_context_reg(ctx_cs, R_0287FC_GE_MAX_OUTPUT_PER_SUBGROUP,
+                              S_0287FC_MAX_VERTS_PER_SUBGROUP(ngg_state->max_out_verts));
+       radeon_set_context_reg(ctx_cs, R_028B4C_GE_NGG_SUBGRP_CNTL,
+                              S_028B4C_PRIM_AMP_FACTOR(ngg_state->prim_amp_factor) |
+                              S_028B4C_THDS_PER_SUBGRP(0)); /* for fast launch */
+       radeon_set_context_reg(ctx_cs, R_028B90_VGT_GS_INSTANCE_CNT,
+                              S_028B90_CNT(gs_num_invocations) |
+                              S_028B90_ENABLE(gs_num_invocations > 1) |
+                              S_028B90_EN_MAX_VERT_OUT_PER_GS_INSTANCE(ngg_state->max_vert_out_per_gs_instance));
+
+       /* User edge flags are set by the pos exports. If user edge flags are
+        * not used, we must use hw-generated edge flags and pass them via
+        * the prim export to prevent drawing lines on internal edges of
+        * decomposed primitives (such as quads) with polygon mode = lines.
+        *
+        * TODO: We should combine hw-generated edge flags with user edge
+        *       flags in the shader.
+        */
+       radeon_set_context_reg(ctx_cs, R_028838_PA_CL_NGG_CNTL,
+                              S_028838_INDEX_BUF_EDGE_FLAG_ENA(!radv_pipeline_has_tess(pipeline) &&
+                                                               !radv_pipeline_has_gs(pipeline)));
+
+       radeon_set_context_reg(ctx_cs, R_03096C_GE_CNTL,
+                              S_03096C_PRIM_GRP_SIZE(ngg_state->max_gsprims) |
+                              S_03096C_VERT_GRP_SIZE(ngg_state->hw_max_esverts) |
+                              S_03096C_BREAK_WAVE_AT_EOI(break_wave_at_eoi));
+}
+
 static void
 radv_pipeline_generate_hw_hs(struct radeon_cmdbuf *cs,
                             struct radv_pipeline *pipeline,
@@ -3082,20 +3431,33 @@ radv_pipeline_generate_hw_hs(struct radeon_cmdbuf *cs,
        uint64_t va = radv_buffer_get_va(shader->bo) + shader->bo_offset;
 
        if (pipeline->device->physical_device->rad_info.chip_class >= GFX9) {
-               radeon_set_sh_reg_seq(cs, R_00B410_SPI_SHADER_PGM_LO_LS, 2);
-               radeon_emit(cs, va >> 8);
-               radeon_emit(cs, S_00B414_MEM_BASE(va >> 40));
+               unsigned hs_rsrc2 = shader->config.rsrc2;
+
+               if (pipeline->device->physical_device->rad_info.chip_class >= GFX10) {
+                       hs_rsrc2 |= S_00B42C_LDS_SIZE_GFX10(tess->lds_size);
+               } else {
+                       hs_rsrc2 |= S_00B42C_LDS_SIZE_GFX9(tess->lds_size);
+               }
+
+               if (pipeline->device->physical_device->rad_info.chip_class >= GFX10) {
+                       radeon_set_sh_reg_seq(cs, R_00B520_SPI_SHADER_PGM_LO_LS, 2);
+                       radeon_emit(cs, va >> 8);
+                       radeon_emit(cs, S_00B524_MEM_BASE(va >> 40));
+               } else {
+                       radeon_set_sh_reg_seq(cs, R_00B410_SPI_SHADER_PGM_LO_LS, 2);
+                       radeon_emit(cs, va >> 8);
+                       radeon_emit(cs, S_00B414_MEM_BASE(va >> 40));
+               }
 
                radeon_set_sh_reg_seq(cs, R_00B428_SPI_SHADER_PGM_RSRC1_HS, 2);
-               radeon_emit(cs, shader->rsrc1);
-               radeon_emit(cs, shader->rsrc2 |
-                               S_00B42C_LDS_SIZE(tess->lds_size));
+               radeon_emit(cs, shader->config.rsrc1);
+               radeon_emit(cs, hs_rsrc2);
        } else {
                radeon_set_sh_reg_seq(cs, R_00B420_SPI_SHADER_PGM_LO_HS, 4);
                radeon_emit(cs, va >> 8);
                radeon_emit(cs, S_00B424_MEM_BASE(va >> 40));
-               radeon_emit(cs, shader->rsrc1);
-               radeon_emit(cs, shader->rsrc2);
+               radeon_emit(cs, shader->config.rsrc1);
+               radeon_emit(cs, shader->config.rsrc2);
        }
 }
 
@@ -3103,7 +3465,8 @@ static void
 radv_pipeline_generate_vertex_shader(struct radeon_cmdbuf *ctx_cs,
                                     struct radeon_cmdbuf *cs,
                                     struct radv_pipeline *pipeline,
-                                    const struct radv_tessellation_state *tess)
+                                    const struct radv_tessellation_state *tess,
+                                    const struct radv_ngg_state *ngg)
 {
        struct radv_shader_variant *vs;
 
@@ -3116,6 +3479,8 @@ radv_pipeline_generate_vertex_shader(struct radeon_cmdbuf *ctx_cs,
                radv_pipeline_generate_hw_ls(cs, pipeline, vs, tess);
        else if (vs->info.vs.as_es)
                radv_pipeline_generate_hw_es(cs, pipeline, vs);
+       else if (vs->info.is_ngg)
+               radv_pipeline_generate_hw_ngg(ctx_cs, cs, pipeline, vs, ngg);
        else
                radv_pipeline_generate_hw_vs(ctx_cs, cs, pipeline, vs);
 }
@@ -3208,13 +3573,19 @@ radv_pipeline_generate_geometry_shader(struct radeon_cmdbuf *ctx_cs,
        va = radv_buffer_get_va(gs->bo) + gs->bo_offset;
 
        if (pipeline->device->physical_device->rad_info.chip_class >= GFX9) {
-               radeon_set_sh_reg_seq(cs, R_00B210_SPI_SHADER_PGM_LO_ES, 2);
-               radeon_emit(cs, va >> 8);
-               radeon_emit(cs, S_00B214_MEM_BASE(va >> 40));
+               if (pipeline->device->physical_device->rad_info.chip_class >= GFX10) {
+                       radeon_set_sh_reg_seq(cs, R_00B320_SPI_SHADER_PGM_LO_ES, 2);
+                       radeon_emit(cs, va >> 8);
+                       radeon_emit(cs, S_00B324_MEM_BASE(va >> 40));
+               } else {
+                       radeon_set_sh_reg_seq(cs, R_00B210_SPI_SHADER_PGM_LO_ES, 2);
+                       radeon_emit(cs, va >> 8);
+                       radeon_emit(cs, S_00B214_MEM_BASE(va >> 40));
+               }
 
                radeon_set_sh_reg_seq(cs, R_00B228_SPI_SHADER_PGM_RSRC1_GS, 2);
-               radeon_emit(cs, gs->rsrc1);
-               radeon_emit(cs, gs->rsrc2 | S_00B22C_LDS_SIZE(gs_state->lds_size));
+               radeon_emit(cs, gs->config.rsrc1);
+               radeon_emit(cs, gs->config.rsrc2 | S_00B22C_LDS_SIZE(gs_state->lds_size));
 
                radeon_set_context_reg(ctx_cs, R_028A44_VGT_GS_ONCHIP_CNTL, gs_state->vgt_gs_onchip_cntl);
                radeon_set_context_reg(ctx_cs, R_028A94_VGT_GS_MAX_PRIMS_PER_SUBGROUP, gs_state->vgt_gs_max_prims_per_subgroup);
@@ -3222,8 +3593,8 @@ radv_pipeline_generate_geometry_shader(struct radeon_cmdbuf *ctx_cs,
                radeon_set_sh_reg_seq(cs, R_00B220_SPI_SHADER_PGM_LO_GS, 4);
                radeon_emit(cs, va >> 8);
                radeon_emit(cs, S_00B224_MEM_BASE(va >> 40));
-               radeon_emit(cs, gs->rsrc1);
-               radeon_emit(cs, gs->rsrc2);
+               radeon_emit(cs, gs->config.rsrc1);
+               radeon_emit(cs, gs->config.rsrc2);
        }
 
        radv_pipeline_generate_hw_vs(ctx_cs, cs, pipeline, pipeline->gs_copy_shader);
@@ -3270,7 +3641,6 @@ radv_pipeline_generate_ps_inputs(struct radeon_cmdbuf *ctx_cs,
        }
 
        if (ps->info.info.ps.layer_input ||
-           ps->info.info.ps.uses_input_attachments ||
            ps->info.info.needs_multiview_view_index) {
                unsigned vs_offset = outinfo->vs_output_param_offset[VARYING_SLOT_LAYER];
                if (vs_offset != AC_EXP_PARAM_UNDEFINED)
@@ -3379,8 +3749,8 @@ radv_pipeline_generate_fragment_shader(struct radeon_cmdbuf *ctx_cs,
        radeon_set_sh_reg_seq(cs, R_00B020_SPI_SHADER_PGM_LO_PS, 4);
        radeon_emit(cs, va >> 8);
        radeon_emit(cs, S_00B024_MEM_BASE(va >> 40));
-       radeon_emit(cs, ps->rsrc1);
-       radeon_emit(cs, ps->rsrc2);
+       radeon_emit(cs, ps->config.rsrc1);
+       radeon_emit(cs, ps->config.rsrc2);
 
        radeon_set_context_reg(ctx_cs, R_02880C_DB_SHADER_CONTROL,
                               radv_compute_db_shader_control(pipeline->device,
@@ -3413,7 +3783,8 @@ static void
 radv_pipeline_generate_vgt_vertex_reuse(struct radeon_cmdbuf *ctx_cs,
                                        struct radv_pipeline *pipeline)
 {
-       if (pipeline->device->physical_device->rad_info.family < CHIP_POLARIS10)
+       if (pipeline->device->physical_device->rad_info.family < CHIP_POLARIS10 ||
+           pipeline->device->physical_device->rad_info.chip_class >= GFX10)
                return;
 
        unsigned vtx_reuse_depth = 30;
@@ -3437,13 +3808,20 @@ radv_compute_vgt_shader_stages_en(const struct radv_pipeline *pipeline)
                        stages |=  S_028B54_ES_EN(V_028B54_ES_STAGE_DS) |
                                S_028B54_GS_EN(1) |
                                S_028B54_VS_EN(V_028B54_VS_STAGE_COPY_SHADER);
+               else if (radv_pipeline_has_ngg(pipeline))
+                       stages |= S_028B54_ES_EN(V_028B54_ES_STAGE_DS) |
+                                 S_028B54_PRIMGEN_EN(1);
                else
                        stages |= S_028B54_VS_EN(V_028B54_VS_STAGE_DS);
 
-       } else if (radv_pipeline_has_gs(pipeline))
+       } else if (radv_pipeline_has_gs(pipeline)) {
                stages |= S_028B54_ES_EN(V_028B54_ES_STAGE_REAL) |
                        S_028B54_GS_EN(1) |
                        S_028B54_VS_EN(V_028B54_VS_STAGE_COPY_SHADER);
+       } else if (radv_pipeline_has_ngg(pipeline)) {
+               stages |= S_028B54_ES_EN(V_028B54_ES_STAGE_REAL) |
+                         S_028B54_PRIMGEN_EN(1);
+       }
 
        if (pipeline->device->physical_device->rad_info.chip_class >= GFX9)
                stages |= S_028B54_MAX_PRIMGRP_IN_WAVE(2);
@@ -3482,6 +3860,41 @@ radv_compute_cliprect_rule(const VkGraphicsPipelineCreateInfo *pCreateInfo)
        return mask;
 }
 
+static void
+gfx10_pipeline_generate_ge_cntl(struct radeon_cmdbuf *ctx_cs,
+                               struct radv_pipeline *pipeline,
+                               const struct radv_tessellation_state *tess,
+                               const struct radv_gs_state *gs_state)
+{
+       bool break_wave_at_eoi = false;
+       unsigned primgroup_size;
+       unsigned vertgroup_size;
+
+       if (radv_pipeline_has_tess(pipeline)) {
+               primgroup_size = tess->num_patches; /* must be a multiple of NUM_PATCHES */
+               vertgroup_size = 0;
+       } else if (radv_pipeline_has_gs(pipeline)) {
+               unsigned vgt_gs_onchip_cntl = gs_state->vgt_gs_onchip_cntl;
+               primgroup_size = G_028A44_GS_PRIMS_PER_SUBGRP(vgt_gs_onchip_cntl);
+               vertgroup_size = G_028A44_ES_VERTS_PER_SUBGRP(vgt_gs_onchip_cntl);
+       } else {
+               primgroup_size = 128; /* recommended without a GS and tess */
+               vertgroup_size = 0;
+       }
+
+       if (radv_pipeline_has_tess(pipeline)) {
+               if (pipeline->shaders[MESA_SHADER_TESS_CTRL]->info.info.uses_prim_id ||
+                   radv_get_shader(pipeline, MESA_SHADER_TESS_EVAL)->info.info.uses_prim_id)
+                       break_wave_at_eoi = true;
+       }
+
+       radeon_set_uconfig_reg(ctx_cs, R_03096C_GE_CNTL,
+                              S_03096C_PRIM_GRP_SIZE(primgroup_size) |
+                              S_03096C_VERT_GRP_SIZE(vertgroup_size) |
+                              S_03096C_PACKET_TO_ONE_PA(0) /* line stipple */ |
+                              S_03096C_BREAK_WAVE_AT_EOI(break_wave_at_eoi));
+}
+
 static void
 radv_pipeline_generate_pm4(struct radv_pipeline *pipeline,
                            const VkGraphicsPipelineCreateInfo *pCreateInfo,
@@ -3489,6 +3902,7 @@ radv_pipeline_generate_pm4(struct radv_pipeline *pipeline,
                            const struct radv_blend_state *blend,
                            const struct radv_tessellation_state *tess,
                            const struct radv_gs_state *gs,
+                           const struct radv_ngg_state *ngg,
                            unsigned prim, unsigned gs_out)
 {
        struct radeon_cmdbuf *ctx_cs = &pipeline->ctx_cs;
@@ -3504,7 +3918,7 @@ radv_pipeline_generate_pm4(struct radv_pipeline *pipeline,
        radv_pipeline_generate_raster_state(ctx_cs, pipeline, pCreateInfo);
        radv_pipeline_generate_multisample_state(ctx_cs, pipeline);
        radv_pipeline_generate_vgt_gs_mode(ctx_cs, pipeline);
-       radv_pipeline_generate_vertex_shader(ctx_cs, cs, pipeline, tess);
+       radv_pipeline_generate_vertex_shader(ctx_cs, cs, pipeline, tess, ngg);
        radv_pipeline_generate_tess_shaders(ctx_cs, cs, pipeline, tess);
        radv_pipeline_generate_geometry_shader(ctx_cs, cs, pipeline, gs);
        radv_pipeline_generate_fragment_shader(ctx_cs, cs, pipeline);
@@ -3512,6 +3926,9 @@ radv_pipeline_generate_pm4(struct radv_pipeline *pipeline,
        radv_pipeline_generate_vgt_vertex_reuse(ctx_cs, pipeline);
        radv_pipeline_generate_binning_state(ctx_cs, pipeline, pCreateInfo);
 
+       if (pipeline->device->physical_device->rad_info.chip_class >= GFX10 && !radv_pipeline_has_ngg(pipeline))
+               gfx10_pipeline_generate_ge_cntl(ctx_cs, pipeline, tess, gs);
+
        radeon_set_context_reg(ctx_cs, R_0286E8_SPI_TMPRING_SIZE,
                               S_0286E8_WAVES(pipeline->max_waves) |
                               S_0286E8_WAVESIZE(pipeline->scratch_bytes_per_wave >> 10));
@@ -3519,7 +3936,8 @@ radv_pipeline_generate_pm4(struct radv_pipeline *pipeline,
        radeon_set_context_reg(ctx_cs, R_028B54_VGT_SHADER_STAGES_EN, radv_compute_vgt_shader_stages_en(pipeline));
 
        if (pipeline->device->physical_device->rad_info.chip_class >= GFX7) {
-               radeon_set_uconfig_reg_idx(cs, R_030908_VGT_PRIMITIVE_TYPE, 1, prim);
+               radeon_set_uconfig_reg_idx(pipeline->device->physical_device,
+                                          cs, R_030908_VGT_PRIMITIVE_TYPE, 1, prim);
        } else {
                radeon_set_config_reg(cs, R_008958_VGT_PRIMITIVE_TYPE, prim);
        }
@@ -3739,6 +4157,9 @@ radv_pipeline_init(struct radv_pipeline *pipeline,
        if (radv_pipeline_has_gs(pipeline)) {
                gs_out = si_conv_gl_prim_to_gs_out(pipeline->shaders[MESA_SHADER_GEOMETRY]->info.gs.output_prim);
                pipeline->graphics.can_use_guardband = gs_out == V_028A6C_OUTPRIM_TYPE_TRISTRIP;
+       } else if (radv_pipeline_has_tess(pipeline)) {
+               gs_out = si_conv_gl_prim_to_gs_out(pipeline->shaders[MESA_SHADER_TESS_EVAL]->info.tes.primitive_mode);
+               pipeline->graphics.can_use_guardband = gs_out == V_028A6C_OUTPRIM_TYPE_TRISTRIP;
        } else {
                gs_out = si_conv_prim_to_gs_out(pCreateInfo->pInputAssemblyState->topology);
        }
@@ -3746,6 +4167,8 @@ radv_pipeline_init(struct radv_pipeline *pipeline,
                prim = V_008958_DI_PT_RECTLIST;
                gs_out = V_028A6C_OUTPRIM_TYPE_TRISTRIP;
                pipeline->graphics.can_use_guardband = true;
+               if (radv_pipeline_has_ngg(pipeline))
+                       gs_out = V_028A6C_VGT_OUT_RECT_V0;
        }
        pipeline->graphics.prim_restart_enable = !!pCreateInfo->pInputAssemblyState->primitiveRestartEnable;
        /* prim vertex count will need TESS changes */
@@ -3778,8 +4201,12 @@ radv_pipeline_init(struct radv_pipeline *pipeline,
                }
        }
 
+       struct radv_ngg_state ngg = {0};
        struct radv_gs_state gs = {0};
-       if (radv_pipeline_has_gs(pipeline)) {
+
+       if (radv_pipeline_has_ngg(pipeline)) {
+               ngg = calculate_ngg_info(pCreateInfo, pipeline);
+       } else if (radv_pipeline_has_gs(pipeline)) {
                gs = calculate_gs_info(pCreateInfo, pipeline);
                calculate_gs_ring_sizes(pipeline, &gs);
        }
@@ -3815,7 +4242,7 @@ radv_pipeline_init(struct radv_pipeline *pipeline,
        pipeline->streamout_shader = radv_pipeline_get_streamout_shader(pipeline);
 
        result = radv_pipeline_scratch_init(device, pipeline);
-       radv_pipeline_generate_pm4(pipeline, pCreateInfo, extra, &blend, &tess, &gs, prim, gs_out);
+       radv_pipeline_generate_pm4(pipeline, pCreateInfo, extra, &blend, &tess, &gs, &ngg, prim, gs_out);
 
        return result;
 }
@@ -3898,8 +4325,8 @@ radv_compute_generate_pm4(struct radv_pipeline *pipeline)
        radeon_emit(&pipeline->cs, S_00B834_DATA(va >> 40));
 
        radeon_set_sh_reg_seq(&pipeline->cs, R_00B848_COMPUTE_PGM_RSRC1, 2);
-       radeon_emit(&pipeline->cs, compute_shader->rsrc1);
-       radeon_emit(&pipeline->cs, compute_shader->rsrc2);
+       radeon_emit(&pipeline->cs, compute_shader->config.rsrc1);
+       radeon_emit(&pipeline->cs, compute_shader->config.rsrc2);
 
        radeon_set_sh_reg(&pipeline->cs, R_00B860_COMPUTE_TMPRING_SIZE,
                          S_00B860_WAVES(pipeline->max_waves) |