radeonsi: move VGT_GS_OUT_PRIM_TYPE into si_shader_gs
[mesa.git] / src / gallium / drivers / radeonsi / si_state_draw.c
index 7e86c0f07dc60cc98408d10e7f978b365e58e45d..845ec468aec0d0a10abc56a316ef0615fee37d96 100644 (file)
@@ -1,5 +1,6 @@
 /*
  * Copyright 2012 Advanced Micro Devices, Inc.
+ * All Rights Reserved.
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -21,9 +22,7 @@
  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
-#include "si_pipe.h"
-#include "radeon/r600_cs.h"
-#include "sid.h"
+#include "si_build_pm4.h"
 #include "gfx9d.h"
 
 #include "util/u_index_modify.h"
@@ -60,31 +59,6 @@ static unsigned si_conv_pipe_prim(unsigned mode)
        return prim_conv[mode];
 }
 
-static unsigned si_conv_prim_to_gs_out(unsigned mode)
-{
-       static const int prim_conv[] = {
-               [PIPE_PRIM_POINTS]                      = V_028A6C_OUTPRIM_TYPE_POINTLIST,
-               [PIPE_PRIM_LINES]                       = V_028A6C_OUTPRIM_TYPE_LINESTRIP,
-               [PIPE_PRIM_LINE_LOOP]                   = V_028A6C_OUTPRIM_TYPE_LINESTRIP,
-               [PIPE_PRIM_LINE_STRIP]                  = V_028A6C_OUTPRIM_TYPE_LINESTRIP,
-               [PIPE_PRIM_TRIANGLES]                   = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_TRIANGLE_STRIP]              = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_TRIANGLE_FAN]                = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_QUADS]                       = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_QUAD_STRIP]                  = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_POLYGON]                     = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_LINES_ADJACENCY]             = V_028A6C_OUTPRIM_TYPE_LINESTRIP,
-               [PIPE_PRIM_LINE_STRIP_ADJACENCY]        = V_028A6C_OUTPRIM_TYPE_LINESTRIP,
-               [PIPE_PRIM_TRIANGLES_ADJACENCY]         = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_TRIANGLE_STRIP_ADJACENCY]    = V_028A6C_OUTPRIM_TYPE_TRISTRIP,
-               [PIPE_PRIM_PATCHES]                     = V_028A6C_OUTPRIM_TYPE_POINTLIST,
-               [SI_PRIM_RECTANGLE_LIST]                = V_028A6C_OUTPRIM_TYPE_TRISTRIP
-       };
-       assert(mode < ARRAY_SIZE(prim_conv));
-
-       return prim_conv[mode];
-}
-
 /**
  * This calculates the LDS size for tessellation shaders (VS, TCS, TES).
  * LS.LDS_SIZE is shared by all 3 shader stages.
@@ -96,7 +70,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
                                       const struct pipe_draw_info *info,
                                       unsigned *num_patches)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
        struct si_shader *ls_current;
        struct si_shader_selector *ls;
        /* The TES pointer will only be used for sctx->last_tcs.
@@ -104,8 +78,8 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
        struct si_shader_selector *tcs =
                sctx->tcs_shader.cso ? sctx->tcs_shader.cso : sctx->tes_shader.cso;
        unsigned tess_uses_primid = sctx->ia_multi_vgt_param_key.u.tess_uses_prim_id;
-       bool has_primid_instancing_bug = sctx->b.chip_class == SI &&
-                                        sctx->b.screen->info.max_se == 1;
+       bool has_primid_instancing_bug = sctx->chip_class == SI &&
+                                        sctx->screen->info.max_se == 1;
        unsigned tes_sh_base = sctx->shader_pointers.sh_base[PIPE_SHADER_TESS_EVAL];
        unsigned num_tcs_input_cp = info->vertices_per_patch;
        unsigned num_tcs_output_cp, num_tcs_inputs, num_tcs_outputs;
@@ -117,7 +91,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
        unsigned offchip_layout, hardware_lds_size, ls_hs_config;
 
        /* Since GFX9 has merged LS-HS in the TCS state, set LS = TCS. */
-       if (sctx->b.chip_class >= GFX9) {
+       if (sctx->chip_class >= GFX9) {
                if (sctx->tcs_shader.cso)
                        ls_current = sctx->tcs_shader.current;
                else
@@ -195,7 +169,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
         */
        *num_patches = MIN2(*num_patches, 40);
 
-       if (sctx->b.chip_class == SI) {
+       if (sctx->chip_class == SI) {
                /* SI bug workaround, related to power management. Limit LS-HS
                 * threadgroups to only one wave.
                 */
@@ -248,7 +222,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
        /* Compute the LDS size. */
        lds_size = output_patch0_offset + output_patch_size * *num_patches;
 
-       if (sctx->b.chip_class >= CIK) {
+       if (sctx->chip_class >= CIK) {
                assert(lds_size <= 65536);
                lds_size = align(lds_size, 512) / 512;
        } else {
@@ -261,7 +235,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
                                  C_VS_STATE_LS_OUT_VERTEX_SIZE;
        sctx->current_vs_state |= tcs_in_layout;
 
-       if (sctx->b.chip_class >= GFX9) {
+       if (sctx->chip_class >= GFX9) {
                unsigned hs_rsrc2 = ls_current->config.rsrc2 |
                                    S_00B42C_LDS_SIZE(lds_size);
 
@@ -282,7 +256,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
 
                /* Due to a hw bug, RSRC2_LS must be written twice with another
                 * LS register written in between. */
-               if (sctx->b.chip_class == CIK && sctx->b.family != CHIP_HAWAII)
+               if (sctx->chip_class == CIK && sctx->family != CHIP_HAWAII)
                        radeon_set_sh_reg(cs, R_00B52C_SPI_SHADER_PGM_RSRC2_LS, ls_rsrc2);
                radeon_set_sh_reg_seq(cs, R_00B528_SPI_SHADER_PGM_RSRC1_LS, 2);
                radeon_emit(cs, ls_current->config.rsrc1);
@@ -306,7 +280,7 @@ static void si_emit_derived_tess_state(struct si_context *sctx,
                       S_028B58_HS_NUM_INPUT_CP(num_tcs_input_cp) |
                       S_028B58_HS_NUM_OUTPUT_CP(num_tcs_output_cp);
 
-       if (sctx->b.chip_class >= CIK)
+       if (sctx->chip_class >= CIK)
                radeon_set_context_reg_idx(cs, R_028B58_VGT_LS_HS_CONFIG, 2,
                                           ls_hs_config);
        else
@@ -363,7 +337,8 @@ si_get_init_multi_vgt_param(struct si_screen *sscreen,
                                    sscreen->info.family == CHIP_FIJI ||
                                    sscreen->info.family == CHIP_POLARIS10 ||
                                    sscreen->info.family == CHIP_POLARIS11 ||
-                                   sscreen->info.family == CHIP_POLARIS12)
+                                   sscreen->info.family == CHIP_POLARIS12 ||
+                                   sscreen->info.family == CHIP_VEGAM)
                                        partial_vs_wave = true;
                        } else {
                                partial_vs_wave = true;
@@ -512,7 +487,7 @@ static unsigned si_get_ia_multi_vgt_param(struct si_context *sctx,
 
        if (sctx->gs_shader.cso) {
                /* GS requirement. */
-               if (sctx->b.chip_class <= VI &&
+               if (sctx->chip_class <= VI &&
                    SI_GS_PER_ES / primgroup_size >= sctx->screen->gs_table_depth - 3)
                        ia_multi_vgt_param |= S_028AA8_PARTIAL_ES_WAVE_ON(1);
 
@@ -520,13 +495,13 @@ static unsigned si_get_ia_multi_vgt_param(struct si_context *sctx,
                 * The hw doc says all multi-SE chips are affected, but Vulkan
                 * only applies it to Hawaii. Do what Vulkan does.
                 */
-               if (sctx->b.family == CHIP_HAWAII &&
+               if (sctx->family == CHIP_HAWAII &&
                    G_028AA8_SWITCH_ON_EOI(ia_multi_vgt_param) &&
                    (info->indirect ||
                     (info->instance_count > 1 &&
                      (info->count_from_stream_output ||
                       si_num_prims_for_vertices(info) <= 1))))
-                       sctx->b.flags |= SI_CONTEXT_VGT_FLUSH;
+                       sctx->flags |= SI_CONTEXT_VGT_FLUSH;
        }
 
        return ia_multi_vgt_param;
@@ -535,7 +510,7 @@ static unsigned si_get_ia_multi_vgt_param(struct si_context *sctx,
 /* rast_prim is the primitive type after GS. */
 static void si_emit_rasterizer_prim_state(struct si_context *sctx)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
        enum pipe_prim_type rast_prim = sctx->current_rast_prim;
        struct si_state_rasterizer *rs = sctx->emitted.named.rasterizer;
 
@@ -575,7 +550,7 @@ static void si_emit_vs_state(struct si_context *sctx,
        }
 
        if (sctx->current_vs_state != sctx->last_vs_state) {
-               struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+               struct radeon_winsys_cs *cs = sctx->gfx_cs;
 
                radeon_set_sh_reg(cs,
                        sctx->shader_pointers.sh_base[PIPE_SHADER_VERTEX] +
@@ -590,18 +565,17 @@ static void si_emit_draw_registers(struct si_context *sctx,
                                   const struct pipe_draw_info *info,
                                   unsigned num_patches)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
        unsigned prim = si_conv_pipe_prim(info->mode);
-       unsigned gs_out_prim = si_conv_prim_to_gs_out(sctx->current_rast_prim);
        unsigned ia_multi_vgt_param;
 
        ia_multi_vgt_param = si_get_ia_multi_vgt_param(sctx, info, num_patches);
 
        /* Draw state. */
        if (ia_multi_vgt_param != sctx->last_multi_vgt_param) {
-               if (sctx->b.chip_class >= GFX9)
+               if (sctx->chip_class >= GFX9)
                        radeon_set_uconfig_reg_idx(cs, R_030960_IA_MULTI_VGT_PARAM, 4, ia_multi_vgt_param);
-               else if (sctx->b.chip_class >= CIK)
+               else if (sctx->chip_class >= CIK)
                        radeon_set_context_reg_idx(cs, R_028AA8_IA_MULTI_VGT_PARAM, 1, ia_multi_vgt_param);
                else
                        radeon_set_context_reg(cs, R_028AA8_IA_MULTI_VGT_PARAM, ia_multi_vgt_param);
@@ -609,7 +583,7 @@ static void si_emit_draw_registers(struct si_context *sctx,
                sctx->last_multi_vgt_param = ia_multi_vgt_param;
        }
        if (prim != sctx->last_prim) {
-               if (sctx->b.chip_class >= CIK)
+               if (sctx->chip_class >= CIK)
                        radeon_set_uconfig_reg_idx(cs, R_030908_VGT_PRIMITIVE_TYPE, 1, prim);
                else
                        radeon_set_config_reg(cs, R_008958_VGT_PRIMITIVE_TYPE, prim);
@@ -617,14 +591,9 @@ static void si_emit_draw_registers(struct si_context *sctx,
                sctx->last_prim = prim;
        }
 
-       if (gs_out_prim != sctx->last_gs_out_prim) {
-               radeon_set_context_reg(cs, R_028A6C_VGT_GS_OUT_PRIM_TYPE, gs_out_prim);
-               sctx->last_gs_out_prim = gs_out_prim;
-       }
-
        /* Primitive restart. */
        if (info->primitive_restart != sctx->last_primitive_restart_en) {
-               if (sctx->b.chip_class >= GFX9)
+               if (sctx->chip_class >= GFX9)
                        radeon_set_uconfig_reg(cs, R_03092C_VGT_MULTI_PRIM_IB_RESET_EN,
                                               info->primitive_restart);
                else
@@ -650,9 +619,9 @@ static void si_emit_draw_packets(struct si_context *sctx,
                                 unsigned index_offset)
 {
        struct pipe_draw_indirect_info *indirect = info->indirect;
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
        unsigned sh_base_reg = sctx->shader_pointers.sh_base[PIPE_SHADER_VERTEX];
-       bool render_cond_bit = sctx->b.render_cond && !sctx->b.render_cond_force_off;
+       bool render_cond_bit = sctx->render_cond && !sctx->render_cond_force_off;
        uint32_t index_max_size = 0;
        uint64_t index_va = 0;
 
@@ -674,7 +643,7 @@ static void si_emit_draw_packets(struct si_context *sctx,
                radeon_emit(cs, R_028B2C_VGT_STRMOUT_DRAW_OPAQUE_BUFFER_FILLED_SIZE >> 2);
                radeon_emit(cs, 0); /* unused */
 
-               radeon_add_to_buffer_list(&sctx->b, sctx->b.gfx_cs,
+               radeon_add_to_buffer_list(sctx, sctx->gfx_cs,
                                      t->buf_filled_size, RADEON_USAGE_READ,
                                      RADEON_PRIO_SO_FILLED_SIZE);
        }
@@ -691,12 +660,12 @@ static void si_emit_draw_packets(struct si_context *sctx,
                                break;
                        case 2:
                                index_type = V_028A7C_VGT_INDEX_16 |
-                                            (SI_BIG_ENDIAN && sctx->b.chip_class <= CIK ?
+                                            (SI_BIG_ENDIAN && sctx->chip_class <= CIK ?
                                                      V_028A7C_VGT_DMA_SWAP_16_BIT : 0);
                                break;
                        case 4:
                                index_type = V_028A7C_VGT_INDEX_32 |
-                                            (SI_BIG_ENDIAN && sctx->b.chip_class <= CIK ?
+                                            (SI_BIG_ENDIAN && sctx->chip_class <= CIK ?
                                                      V_028A7C_VGT_DMA_SWAP_32_BIT : 0);
                                break;
                        default:
@@ -704,7 +673,7 @@ static void si_emit_draw_packets(struct si_context *sctx,
                                return;
                        }
 
-                       if (sctx->b.chip_class >= GFX9) {
+                       if (sctx->chip_class >= GFX9) {
                                radeon_set_uconfig_reg_idx(cs, R_03090C_VGT_INDEX_TYPE,
                                                           2, index_type);
                        } else {
@@ -719,14 +688,14 @@ static void si_emit_draw_packets(struct si_context *sctx,
                                  index_size;
                index_va = r600_resource(indexbuf)->gpu_address + index_offset;
 
-               radeon_add_to_buffer_list(&sctx->b, sctx->b.gfx_cs,
-                                     (struct r600_resource *)indexbuf,
+               radeon_add_to_buffer_list(sctx, sctx->gfx_cs,
+                                     r600_resource(indexbuf),
                                      RADEON_USAGE_READ, RADEON_PRIO_INDEX_BUFFER);
        } else {
                /* On CI and later, non-indexed draws overwrite VGT_INDEX_TYPE,
                 * so the state must be re-emitted before the next indexed draw.
                 */
-               if (sctx->b.chip_class >= CIK)
+               if (sctx->chip_class >= CIK)
                        sctx->last_index_size = -1;
        }
 
@@ -742,8 +711,8 @@ static void si_emit_draw_packets(struct si_context *sctx,
                radeon_emit(cs, indirect_va);
                radeon_emit(cs, indirect_va >> 32);
 
-               radeon_add_to_buffer_list(&sctx->b, sctx->b.gfx_cs,
-                                     (struct r600_resource *)indirect->buffer,
+               radeon_add_to_buffer_list(sctx, sctx->gfx_cs,
+                                     r600_resource(indirect->buffer),
                                      RADEON_USAGE_READ, RADEON_PRIO_DRAW_INDIRECT);
 
                unsigned di_src_sel = index_size ? V_0287F0_DI_SRC_SEL_DMA
@@ -773,10 +742,10 @@ static void si_emit_draw_packets(struct si_context *sctx,
 
                        if (indirect->indirect_draw_count) {
                                struct r600_resource *params_buf =
-                                       (struct r600_resource *)indirect->indirect_draw_count;
+                                       r600_resource(indirect->indirect_draw_count);
 
                                radeon_add_to_buffer_list(
-                                       &sctx->b, sctx->b.gfx_cs, params_buf,
+                                       sctx, sctx->gfx_cs, params_buf,
                                        RADEON_USAGE_READ, RADEON_PRIO_DRAW_INDIRECT);
 
                                count_va = params_buf->gpu_address + indirect->indirect_draw_count_offset;
@@ -852,9 +821,9 @@ static void si_emit_draw_packets(struct si_context *sctx,
 static void si_emit_surface_sync(struct si_context *sctx,
                                 unsigned cp_coher_cntl)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
 
-       if (sctx->b.chip_class >= GFX9) {
+       if (sctx->chip_class >= GFX9) {
                /* Flush caches and wait for the caches to assert idle. */
                radeon_emit(cs, PKT3(PKT3_ACQUIRE_MEM, 5, 0));
                radeon_emit(cs, cp_coher_cntl); /* CP_COHER_CNTL */
@@ -875,16 +844,16 @@ static void si_emit_surface_sync(struct si_context *sctx,
 
 void si_emit_cache_flush(struct si_context *sctx)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
-       uint32_t flags = sctx->b.flags;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
+       uint32_t flags = sctx->flags;
        uint32_t cp_coher_cntl = 0;
        uint32_t flush_cb_db = flags & (SI_CONTEXT_FLUSH_AND_INV_CB |
                                        SI_CONTEXT_FLUSH_AND_INV_DB);
 
        if (flags & SI_CONTEXT_FLUSH_AND_INV_CB)
-               sctx->b.num_cb_cache_flushes++;
+               sctx->num_cb_cache_flushes++;
        if (flags & SI_CONTEXT_FLUSH_AND_INV_DB)
-               sctx->b.num_db_cache_flushes++;
+               sctx->num_db_cache_flushes++;
 
        /* SI has a bug that it always flushes ICACHE and KCACHE if either
         * bit is set. An alternative way is to write SQC_CACHES, but that
@@ -899,7 +868,7 @@ void si_emit_cache_flush(struct si_context *sctx)
        if (flags & SI_CONTEXT_INV_SMEM_L1)
                cp_coher_cntl |= S_0085F0_SH_KCACHE_ACTION_ENA(1);
 
-       if (sctx->b.chip_class <= VI) {
+       if (sctx->chip_class <= VI) {
                if (flags & SI_CONTEXT_FLUSH_AND_INV_CB) {
                        cp_coher_cntl |= S_0085F0_CB_ACTION_ENA(1) |
                                         S_0085F0_CB0_DEST_BASE_ENA(1) |
@@ -912,8 +881,8 @@ void si_emit_cache_flush(struct si_context *sctx)
                                         S_0085F0_CB7_DEST_BASE_ENA(1);
 
                        /* Necessary for DCC */
-                       if (sctx->b.chip_class == VI)
-                               si_gfx_write_event_eop(&sctx->b, V_028A90_FLUSH_AND_INV_CB_DATA_TS,
+                       if (sctx->chip_class == VI)
+                               si_gfx_write_event_eop(sctx, V_028A90_FLUSH_AND_INV_CB_DATA_TS,
                                                       0, EOP_DATA_SEL_DISCARD, NULL,
                                                       0, 0, SI_NOT_QUERY);
                }
@@ -945,12 +914,12 @@ void si_emit_cache_flush(struct si_context *sctx)
                        /* Only count explicit shader flushes, not implicit ones
                         * done by SURFACE_SYNC.
                         */
-                       sctx->b.num_vs_flushes++;
-                       sctx->b.num_ps_flushes++;
+                       sctx->num_vs_flushes++;
+                       sctx->num_ps_flushes++;
                } else if (flags & SI_CONTEXT_VS_PARTIAL_FLUSH) {
                        radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
                        radeon_emit(cs, EVENT_TYPE(V_028A90_VS_PARTIAL_FLUSH) | EVENT_INDEX(4));
-                       sctx->b.num_vs_flushes++;
+                       sctx->num_vs_flushes++;
                }
        }
 
@@ -958,7 +927,7 @@ void si_emit_cache_flush(struct si_context *sctx)
            sctx->compute_is_busy) {
                radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
                radeon_emit(cs, EVENT_TYPE(V_028A90_CS_PARTIAL_FLUSH | EVENT_INDEX(4)));
-               sctx->b.num_cs_flushes++;
+               sctx->num_cs_flushes++;
                sctx->compute_is_busy = false;
        }
 
@@ -975,7 +944,7 @@ void si_emit_cache_flush(struct si_context *sctx)
        /* GFX9: Wait for idle if we're flushing CB or DB. ACQUIRE_MEM doesn't
         * wait for idle on GFX9. We have to use a TS event.
         */
-       if (sctx->b.chip_class >= GFX9 && flush_cb_db) {
+       if (sctx->chip_class >= GFX9 && flush_cb_db) {
                uint64_t va;
                unsigned tc_flags, cb_db_event;
 
@@ -1021,18 +990,18 @@ void si_emit_cache_flush(struct si_context *sctx)
                        flags &= ~(SI_CONTEXT_INV_GLOBAL_L2 |
                                   SI_CONTEXT_WRITEBACK_GLOBAL_L2 |
                                   SI_CONTEXT_INV_VMEM_L1);
-                       sctx->b.num_L2_invalidates++;
+                       sctx->num_L2_invalidates++;
                }
 
                /* Do the flush (enqueue the event and wait for it). */
                va = sctx->wait_mem_scratch->gpu_address;
                sctx->wait_mem_number++;
 
-               si_gfx_write_event_eop(&sctx->b, cb_db_event, tc_flags,
+               si_gfx_write_event_eop(sctx, cb_db_event, tc_flags,
                                       EOP_DATA_SEL_VALUE_32BIT,
                                       sctx->wait_mem_scratch, va,
                                       sctx->wait_mem_number, SI_NOT_QUERY);
-               si_gfx_wait_fence(&sctx->b, va, sctx->wait_mem_number, 0xffffffff);
+               si_gfx_wait_fence(sctx, va, sctx->wait_mem_number, 0xffffffff);
        }
 
        /* Make sure ME is idle (it executes most packets) before continuing.
@@ -1057,7 +1026,7 @@ void si_emit_cache_flush(struct si_context *sctx)
         * SI-CIK don't support L2 write-back.
         */
        if (flags & SI_CONTEXT_INV_GLOBAL_L2 ||
-           (sctx->b.chip_class <= CIK &&
+           (sctx->chip_class <= CIK &&
             (flags & SI_CONTEXT_WRITEBACK_GLOBAL_L2))) {
                /* Invalidate L1 & L2. (L1 is always invalidated on SI)
                 * WB must be set on VI+ when TC_ACTION is set.
@@ -1065,9 +1034,9 @@ void si_emit_cache_flush(struct si_context *sctx)
                si_emit_surface_sync(sctx, cp_coher_cntl |
                                     S_0085F0_TC_ACTION_ENA(1) |
                                     S_0085F0_TCL1_ACTION_ENA(1) |
-                                    S_0301F0_TC_WB_ACTION_ENA(sctx->b.chip_class >= VI));
+                                    S_0301F0_TC_WB_ACTION_ENA(sctx->chip_class >= VI));
                cp_coher_cntl = 0;
-               sctx->b.num_L2_invalidates++;
+               sctx->num_L2_invalidates++;
        } else {
                /* L1 invalidation and L2 writeback must be done separately,
                 * because both operations can't be done together.
@@ -1083,7 +1052,7 @@ void si_emit_cache_flush(struct si_context *sctx)
                                             S_0301F0_TC_WB_ACTION_ENA(1) |
                                             S_0301F0_TC_NC_ACTION_ENA(1));
                        cp_coher_cntl = 0;
-                       sctx->b.num_L2_writebacks++;
+                       sctx->num_L2_writebacks++;
                }
                if (flags & SI_CONTEXT_INV_VMEM_L1) {
                        /* Invalidate per-CU VMEM L1. */
@@ -1107,7 +1076,7 @@ void si_emit_cache_flush(struct si_context *sctx)
                                EVENT_INDEX(0));
        }
 
-       sctx->b.flags = 0;
+       sctx->flags = 0;
 }
 
 static void si_get_draw_start_count(struct si_context *sctx,
@@ -1124,7 +1093,7 @@ static void si_get_draw_start_count(struct si_context *sctx,
                unsigned *data;
 
                if (indirect->indirect_draw_count) {
-                       data = pipe_buffer_map_range(&sctx->b.b,
+                       data = pipe_buffer_map_range(&sctx->b,
                                        indirect->indirect_draw_count,
                                        indirect->indirect_draw_count_offset,
                                        sizeof(unsigned),
@@ -1132,7 +1101,7 @@ static void si_get_draw_start_count(struct si_context *sctx,
 
                        indirect_count = *data;
 
-                       pipe_buffer_unmap(&sctx->b.b, transfer);
+                       pipe_buffer_unmap(&sctx->b, transfer);
                } else {
                        indirect_count = indirect->draw_count;
                }
@@ -1143,7 +1112,7 @@ static void si_get_draw_start_count(struct si_context *sctx,
                }
 
                map_size = (indirect_count - 1) * indirect->stride + 3 * sizeof(unsigned);
-               data = pipe_buffer_map_range(&sctx->b.b, indirect->buffer,
+               data = pipe_buffer_map_range(&sctx->b, indirect->buffer,
                                             indirect->offset, map_size,
                                             PIPE_TRANSFER_READ, &transfer);
 
@@ -1162,7 +1131,7 @@ static void si_get_draw_start_count(struct si_context *sctx,
                        data += indirect->stride / sizeof(unsigned);
                }
 
-               pipe_buffer_unmap(&sctx->b.b, transfer);
+               pipe_buffer_unmap(&sctx->b, transfer);
 
                if (begin < end) {
                        *start = begin;
@@ -1181,11 +1150,9 @@ static void si_emit_all_states(struct si_context *sctx, const struct pipe_draw_i
 {
        /* Emit state atoms. */
        unsigned mask = sctx->dirty_atoms & ~skip_atom_mask;
-       while (mask) {
-               struct r600_atom *atom = sctx->atoms.array[u_bit_scan(&mask)];
+       while (mask)
+               sctx->atoms.array[u_bit_scan(&mask)].emit(sctx);
 
-               atom->emit(sctx, atom);
-       }
        sctx->dirty_atoms &= skip_atom_mask;
 
        /* Emit states. */
@@ -1250,13 +1217,13 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
        }
 
        /* Recompute and re-emit the texture resource states if needed. */
-       dirty_tex_counter = p_atomic_read(&sctx->b.screen->dirty_tex_counter);
-       if (unlikely(dirty_tex_counter != sctx->b.last_dirty_tex_counter)) {
-               sctx->b.last_dirty_tex_counter = dirty_tex_counter;
+       dirty_tex_counter = p_atomic_read(&sctx->screen->dirty_tex_counter);
+       if (unlikely(dirty_tex_counter != sctx->last_dirty_tex_counter)) {
+               sctx->last_dirty_tex_counter = dirty_tex_counter;
                sctx->framebuffer.dirty_cbufs |=
                        ((1 << sctx->framebuffer.state.nr_cbufs) - 1);
                sctx->framebuffer.dirty_zsbuf = true;
-               si_mark_atom_dirty(sctx, &sctx->framebuffer.atom);
+               si_mark_atom_dirty(sctx, &sctx->atoms.s.framebuffer);
                si_update_all_texture_descriptors(sctx);
        }
 
@@ -1282,7 +1249,7 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
                bool new_is_poly = rast_prim >= PIPE_PRIM_TRIANGLES;
                if (old_is_poly != new_is_poly) {
                        sctx->scissors.dirty_mask = (1 << SI_MAX_VIEWPORTS) - 1;
-                       si_mark_atom_dirty(sctx, &sctx->scissors.atom);
+                       si_mark_atom_dirty(sctx, &sctx->atoms.s.scissors);
                }
 
                sctx->current_rast_prim = rast_prim;
@@ -1334,7 +1301,7 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
        if (index_size) {
                /* Translate or upload, if needed. */
                /* 8-bit indices are supported on VI. */
-               if (sctx->b.chip_class <= CIK && index_size == 1) {
+               if (sctx->chip_class <= CIK && index_size == 1) {
                        unsigned start, count, start_offset, size, offset;
                        void *ptr;
 
@@ -1350,7 +1317,7 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
                        if (!indexbuf)
                                return;
 
-                       util_shorten_ubyte_elts_to_userptr(&sctx->b.b, info, 0, 0,
+                       util_shorten_ubyte_elts_to_userptr(&sctx->b, info, 0, 0,
                                                           index_offset + start,
                                                           count, ptr);
 
@@ -1374,11 +1341,11 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
 
                        /* info->start will be added by the drawing code */
                        index_offset -= start_offset;
-               } else if (sctx->b.chip_class <= CIK &&
+               } else if (sctx->chip_class <= CIK &&
                           r600_resource(indexbuf)->TC_L2_dirty) {
                        /* VI reads index buffers through TC L2, so it doesn't
                         * need this. */
-                       sctx->b.flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
+                       sctx->flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
                        r600_resource(indexbuf)->TC_L2_dirty = false;
                }
        }
@@ -1387,18 +1354,18 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
                struct pipe_draw_indirect_info *indirect = info->indirect;
 
                /* Add the buffer size for memory checking in need_cs_space. */
-               si_context_add_resource_size(ctx, indirect->buffer);
+               si_context_add_resource_size(sctx, indirect->buffer);
 
                /* Indirect buffers use TC L2 on GFX9, but not older hw. */
-               if (sctx->b.chip_class <= VI) {
+               if (sctx->chip_class <= VI) {
                        if (r600_resource(indirect->buffer)->TC_L2_dirty) {
-                               sctx->b.flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
+                               sctx->flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
                                r600_resource(indirect->buffer)->TC_L2_dirty = false;
                        }
 
                        if (indirect->indirect_draw_count &&
                            r600_resource(indirect->indirect_draw_count)->TC_L2_dirty) {
-                               sctx->b.flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
+                               sctx->flags |= SI_CONTEXT_WRITEBACK_GLOBAL_L2;
                                r600_resource(indirect->indirect_draw_count)->TC_L2_dirty = false;
                        }
                }
@@ -1406,7 +1373,7 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
 
        si_need_gfx_cs_space(sctx);
 
-       /* Since we've called r600_context_add_resource_size for vertex buffers,
+       /* Since we've called si_context_add_resource_size for vertex buffers,
         * this must be called after si_need_cs_space, because we must let
         * need_cs_space flush before we add buffers to the buffer list.
         */
@@ -1417,43 +1384,37 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
         * scissor registers are changed. There is also a more efficient but
         * more involved alternative workaround.
         */
-       if ((sctx->b.family == CHIP_VEGA10 || sctx->b.family == CHIP_RAVEN) &&
-           si_is_atom_dirty(sctx, &sctx->scissors.atom)) {
-               sctx->b.flags |= SI_CONTEXT_PS_PARTIAL_FLUSH;
+       if ((sctx->family == CHIP_VEGA10 || sctx->family == CHIP_RAVEN) &&
+           si_is_atom_dirty(sctx, &sctx->atoms.s.scissors)) {
+               sctx->flags |= SI_CONTEXT_PS_PARTIAL_FLUSH;
                si_emit_cache_flush(sctx);
        }
 
        /* Use optimal packet order based on whether we need to sync the pipeline. */
-       if (unlikely(sctx->b.flags & (SI_CONTEXT_FLUSH_AND_INV_CB |
+       if (unlikely(sctx->flags & (SI_CONTEXT_FLUSH_AND_INV_CB |
                                      SI_CONTEXT_FLUSH_AND_INV_DB |
                                      SI_CONTEXT_PS_PARTIAL_FLUSH |
                                      SI_CONTEXT_CS_PARTIAL_FLUSH))) {
                /* If we have to wait for idle, set all states first, so that all
                 * SET packets are processed in parallel with previous draw calls.
-                * Then upload descriptors, set shader pointers, and draw, and
-                * prefetch at the end. This ensures that the time the CUs
-                * are idle is very short. (there are only SET_SH packets between
-                * the wait and the draw)
+                * Then draw and prefetch at the end. This ensures that the time
+                * the CUs are idle is very short.
                 */
-               struct r600_atom *shader_pointers = &sctx->shader_pointers.atom;
-               unsigned masked_atoms = 1u << shader_pointers->id;
+               unsigned masked_atoms = 0;
+
+               if (unlikely(sctx->flags & SI_CONTEXT_FLUSH_FOR_RENDER_COND))
+                       masked_atoms |= si_get_atom_bit(sctx, &sctx->atoms.s.render_cond);
 
-               if (unlikely(sctx->b.flags & SI_CONTEXT_FLUSH_FOR_RENDER_COND))
-                       masked_atoms |= 1u << sctx->b.render_cond_atom.id;
+               if (!si_upload_graphics_shader_descriptors(sctx))
+                       return;
 
-               /* Emit all states except shader pointers and render condition. */
+               /* Emit all states except possibly render condition. */
                si_emit_all_states(sctx, info, masked_atoms);
                si_emit_cache_flush(sctx);
-
                /* <-- CUs are idle here. */
-               if (!si_upload_graphics_shader_descriptors(sctx))
-                       return;
 
-               /* Set shader pointers after descriptors are uploaded. */
-               if (si_is_atom_dirty(sctx, shader_pointers))
-                       shader_pointers->emit(sctx, NULL);
-               if (si_is_atom_dirty(sctx, &sctx->b.render_cond_atom))
-                       sctx->b.render_cond_atom.emit(sctx, NULL);
+               if (si_is_atom_dirty(sctx, &sctx->atoms.s.render_cond))
+                       sctx->atoms.s.render_cond.emit(sctx);
                sctx->dirty_atoms = 0;
 
                si_emit_draw_packets(sctx, info, indexbuf, index_size, index_offset);
@@ -1462,49 +1423,55 @@ void si_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info)
                /* Start prefetches after the draw has been started. Both will run
                 * in parallel, but starting the draw first is more important.
                 */
-               if (sctx->b.chip_class >= CIK && sctx->prefetch_L2_mask)
-                       cik_emit_prefetch_L2(sctx);
+               if (sctx->chip_class >= CIK && sctx->prefetch_L2_mask)
+                       cik_emit_prefetch_L2(sctx, false);
        } else {
                /* If we don't wait for idle, start prefetches first, then set
                 * states, and draw at the end.
                 */
-               if (sctx->b.flags)
+               if (sctx->flags)
                        si_emit_cache_flush(sctx);
 
-               if (sctx->b.chip_class >= CIK && sctx->prefetch_L2_mask)
-                       cik_emit_prefetch_L2(sctx);
+               /* Only prefetch the API VS and VBO descriptors. */
+               if (sctx->chip_class >= CIK && sctx->prefetch_L2_mask)
+                       cik_emit_prefetch_L2(sctx, true);
 
                if (!si_upload_graphics_shader_descriptors(sctx))
                        return;
 
                si_emit_all_states(sctx, info, 0);
                si_emit_draw_packets(sctx, info, indexbuf, index_size, index_offset);
+
+               /* Prefetch the remaining shaders after the draw has been
+                * started. */
+               if (sctx->chip_class >= CIK && sctx->prefetch_L2_mask)
+                       cik_emit_prefetch_L2(sctx, false);
        }
 
        if (unlikely(sctx->current_saved_cs)) {
                si_trace_emit(sctx);
-               si_log_draw_state(sctx, sctx->b.log);
+               si_log_draw_state(sctx, sctx->log);
        }
 
        /* Workaround for a VGT hang when streamout is enabled.
         * It must be done after drawing. */
-       if ((sctx->b.family == CHIP_HAWAII ||
-            sctx->b.family == CHIP_TONGA ||
-            sctx->b.family == CHIP_FIJI) &&
+       if ((sctx->family == CHIP_HAWAII ||
+            sctx->family == CHIP_TONGA ||
+            sctx->family == CHIP_FIJI) &&
            si_get_strmout_en(sctx)) {
-               sctx->b.flags |= SI_CONTEXT_VGT_STREAMOUT_SYNC;
+               sctx->flags |= SI_CONTEXT_VGT_STREAMOUT_SYNC;
        }
 
        if (unlikely(sctx->decompression_enabled)) {
-               sctx->b.num_decompress_calls++;
+               sctx->num_decompress_calls++;
        } else {
-               sctx->b.num_draw_calls++;
+               sctx->num_draw_calls++;
                if (sctx->framebuffer.state.nr_cbufs > 1)
-                       sctx->b.num_mrt_draw_calls++;
+                       sctx->num_mrt_draw_calls++;
                if (info->primitive_restart)
-                       sctx->b.num_prim_restart_calls++;
+                       sctx->num_prim_restart_calls++;
                if (G_0286E8_WAVESIZE(sctx->spi_tmpring_size))
-                       sctx->b.num_spill_draw_calls++;
+                       sctx->num_spill_draw_calls++;
        }
        if (index_size && indexbuf != info->index.resource)
                pipe_resource_reference(&indexbuf, NULL);
@@ -1557,7 +1524,7 @@ void si_draw_rectangle(struct blitter_context *blitter,
 
 void si_trace_emit(struct si_context *sctx)
 {
-       struct radeon_winsys_cs *cs = sctx->b.gfx_cs;
+       struct radeon_winsys_cs *cs = sctx->gfx_cs;
        uint64_t va = sctx->current_saved_cs->trace_buf->gpu_address;
        uint32_t trace_id = ++sctx->current_saved_cs->trace_id;
 
@@ -1571,6 +1538,6 @@ void si_trace_emit(struct si_context *sctx)
        radeon_emit(cs, PKT3(PKT3_NOP, 0, 0));
        radeon_emit(cs, AC_ENCODE_TRACE_POINT(trace_id));
 
-       if (sctx->b.log)
-               u_log_flush(sctx->b.log);
+       if (sctx->log)
+               u_log_flush(sctx->log);
 }