[WHY]
These registers would be useful to know when debugging pstate issues.
[HOW]
Add additional registers to hw state query.
Reviewed-by: Aric Cyr <Aric.Cyr@amd.com>
Reviewed-by: Jun Lei <Jun.Lei@amd.com>
Acked-by: Alex Hung <alex.hung@amd.com>
Signed-off-by: Sung Lee <sunglee@amd.com>
Tested-by: Daniel Wheeler <daniel.wheeler@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
        uint32_t primary_surface_addr_hi;
        uint32_t primary_meta_addr_lo;
        uint32_t primary_meta_addr_hi;
+       uint32_t uclk_pstate_force;
+       uint32_t hubp_cntl;
 };
 
 struct dcn10_hubp {
 
                REG_WRITE(DCHUBBUB_TEST_DEBUG_INDEX, 0x6);
                hubbub_state->test_debug_data = REG_READ(DCHUBBUB_TEST_DEBUG_DATA);
        }
+
+       if (REG(DCHUBBUB_ARB_WATERMARK_CHANGE_CNTL))
+               hubbub_state->watermark_change_cntl = REG_READ(DCHUBBUB_ARB_WATERMARK_CHANGE_CNTL);
+
+       if (REG(DCHUBBUB_ARB_DRAM_STATE_CNTL))
+               hubbub_state->dram_state_cntl = REG_READ(DCHUBBUB_ARB_DRAM_STATE_CNTL);
 }
 
 static const struct hubbub_funcs hubbub2_funcs = {
 
                SWATH_HEIGHT_C, &rq_regs->rq_regs_c.swath_height,
                PTE_ROW_HEIGHT_LINEAR_C, &rq_regs->rq_regs_c.pte_row_height_linear);
 
+       if (REG(UCLK_PSTATE_FORCE))
+               s->uclk_pstate_force = REG_READ(UCLK_PSTATE_FORCE);
+
+       if (REG(DCHUBP_CNTL))
+               s->hubp_cntl = REG_READ(DCHUBP_CNTL);
+
 }
 
 void hubp3_setup(
 
        uint32_t vm_error_pipe;
        uint32_t vm_error_mode;
        uint32_t test_debug_data;
+       uint32_t watermark_change_cntl;
+       uint32_t dram_state_cntl;
 };
 
 struct hubbub_funcs {