Also removes PKT3_CONTEXT_CONTROL as that is already being done by si_begin_new_cs, when emitting init_config.
Signed-off-by: Bas Nieuwenhuizen <[email protected]> --- src/gallium/drivers/radeonsi/si_compute.c | 69 +++++++++++++++------------- src/gallium/drivers/radeonsi/si_hw_context.c | 2 + src/gallium/drivers/radeonsi/si_pipe.h | 1 + 3 files changed, 40 insertions(+), 32 deletions(-) diff --git a/src/gallium/drivers/radeonsi/si_compute.c b/src/gallium/drivers/radeonsi/si_compute.c index 9c25335..7320ace 100644 --- a/src/gallium/drivers/radeonsi/si_compute.c +++ b/src/gallium/drivers/radeonsi/si_compute.c @@ -226,6 +226,39 @@ static unsigned compute_num_waves_for_scratch( return scratch_waves; } +static void si_initialize_compute(struct si_context *sctx) +{ + struct radeon_winsys_cs *cs = sctx->b.gfx.cs; + + radeon_emit(cs, PKT3(PKT3_SET_SH_REG, 3, 0)); + radeon_emit(cs, (R_00B810_COMPUTE_START_X - SI_SH_REG_OFFSET) >> 2); + radeon_emit(cs, 0); + radeon_emit(cs, 0); + radeon_emit(cs, 0); + + radeon_emit(cs, PKT3(PKT3_SET_SH_REG, 3, 0)); + radeon_emit(cs, (R_00B854_COMPUTE_RESOURCE_LIMITS - SI_SH_REG_OFFSET) >> 2); + radeon_emit(cs, 0); + radeon_emit(cs, S_00B858_SH0_CU_EN(0xffff) | S_00B858_SH1_CU_EN(0xffff)); + radeon_emit(cs, S_00B85C_SH0_CU_EN(0xffff) | S_00B85C_SH1_CU_EN(0xffff)); + + /* This register has been moved to R_00CD20_COMPUTE_MAX_WAVE_ID + * and is now per pipe, so it should be handled in the + * kernel if we want to use something other than the default value, + * which is now 0x22f. + */ + if (sctx->b.chip_class <= SI) { + /* XXX: This should be: + * (number of compute units) * 4 * (waves per simd) - 1 */ + + radeon_emit(cs, PKT3(PKT3_SET_SH_REG, 1, 0)); + radeon_emit(cs, (R_00B82C_COMPUTE_MAX_WAVE_ID - SI_SH_REG_OFFSET) >> 2); + radeon_emit(cs, 0x190 /* Default value */); + } + + sctx->cs_shader_state.initialized = true; +} + static void si_upload_compute_input(struct si_context *sctx, const struct pipe_grid_info *info) { @@ -281,7 +314,6 @@ static void si_launch_grid( struct pipe_context *ctx, const struct pipe_grid_info *info) { struct si_context *sctx = (struct si_context*)ctx; - struct radeon_winsys_cs *cs = sctx->b.gfx.cs; struct si_compute *program = sctx->cs_shader_state.program; struct si_pm4_state *pm4 = CALLOC_STRUCT(si_pm4_state); uint64_t shader_va; @@ -290,9 +322,10 @@ static void si_launch_grid( unsigned lds_blocks; unsigned num_waves_for_scratch; - radeon_emit(cs, PKT3(PKT3_CONTEXT_CONTROL, 1, 0) | PKT3_SHADER_TYPE_S(1)); - radeon_emit(cs, 0x80000000); - radeon_emit(cs, 0x80000000); + si_need_cs_space(sctx); + + if (!sctx->cs_shader_state.initialized) + si_initialize_compute(sctx); sctx->b.flags |= SI_CONTEXT_INV_VMEM_L1 | SI_CONTEXT_INV_GLOBAL_L2 | @@ -327,10 +360,6 @@ static void si_launch_grid( RADEON_PRIO_SCRATCH_BUFFER); } - si_pm4_set_reg(pm4, R_00B810_COMPUTE_START_X, 0); - si_pm4_set_reg(pm4, R_00B814_COMPUTE_START_Y, 0); - si_pm4_set_reg(pm4, R_00B818_COMPUTE_START_Z, 0); - si_pm4_set_reg(pm4, R_00B81C_COMPUTE_NUM_THREAD_X, S_00B81C_NUM_THREAD_FULL(info->block[0])); si_pm4_set_reg(pm4, R_00B820_COMPUTE_NUM_THREAD_Y, @@ -350,19 +379,6 @@ static void si_launch_grid( RADEON_PRIO_COMPUTE_GLOBAL); } - /* This register has been moved to R_00CD20_COMPUTE_MAX_WAVE_ID - * and is now per pipe, so it should be handled in the - * kernel if we want to use something other than the default value, - * which is now 0x22f. - */ - if (sctx->b.chip_class <= SI) { - /* XXX: This should be: - * (number of compute units) * 4 * (waves per simd) - 1 */ - - si_pm4_set_reg(pm4, R_00B82C_COMPUTE_MAX_WAVE_ID, - 0x190 /* Default value */); - } - shader_va = shader->bo->gpu_address; shader_va += info->pc; @@ -391,17 +407,6 @@ static void si_launch_grid( shader->config.rsrc2 |= S_00B84C_LDS_SIZE(lds_blocks); si_pm4_set_reg(pm4, R_00B84C_COMPUTE_PGM_RSRC2, shader->config.rsrc2); - si_pm4_set_reg(pm4, R_00B854_COMPUTE_RESOURCE_LIMITS, 0); - - si_pm4_set_reg(pm4, R_00B858_COMPUTE_STATIC_THREAD_MGMT_SE0, - S_00B858_SH0_CU_EN(0xffff /* Default value */) - | S_00B858_SH1_CU_EN(0xffff /* Default value */)) - ; - - si_pm4_set_reg(pm4, R_00B85C_COMPUTE_STATIC_THREAD_MGMT_SE1, - S_00B85C_SH0_CU_EN(0xffff /* Default value */) - | S_00B85C_SH1_CU_EN(0xffff /* Default value */)) - ; num_waves_for_scratch = MIN2(num_waves_for_scratch, diff --git a/src/gallium/drivers/radeonsi/si_hw_context.c b/src/gallium/drivers/radeonsi/si_hw_context.c index 8c900a4..b638bf5 100644 --- a/src/gallium/drivers/radeonsi/si_hw_context.c +++ b/src/gallium/drivers/radeonsi/si_hw_context.c @@ -210,4 +210,6 @@ void si_begin_new_cs(struct si_context *ctx) ctx->last_tcs = NULL; ctx->last_tes_sh_base = -1; ctx->last_num_tcs_input_cp = -1; + + ctx->cs_shader_state.initialized = false; } diff --git a/src/gallium/drivers/radeonsi/si_pipe.h b/src/gallium/drivers/radeonsi/si_pipe.h index dfdb8bb..c85418a 100644 --- a/src/gallium/drivers/radeonsi/si_pipe.h +++ b/src/gallium/drivers/radeonsi/si_pipe.h @@ -133,6 +133,7 @@ struct si_sampler_state { struct si_cs_shader_state { struct si_compute *program; + bool initialized; }; struct si_textures_info { -- 2.7.4 _______________________________________________ mesa-dev mailing list [email protected] https://lists.freedesktop.org/mailman/listinfo/mesa-dev
