Signed-off-by: Dmytro Laktyushkin <Dmytro.Laktyushkin@amd.com>
Reviewed-by: Tony Cheng <Tony.Cheng@amd.com>
Acked-by: Harry Wentland <Harry.Wentland@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
                        plane_id,
                        dppclk_div);
 
-       if (dppclk_div) {
-               /* 1/2 DISPCLK*/
+       if (hws->shifts->DPPCLK_RATE_CONTROL)
                REG_UPDATE_2(DPP_CONTROL[plane_id],
-                       DPPCLK_RATE_CONTROL, 1,
+                       DPPCLK_RATE_CONTROL, dppclk_div,
                        DPP_CLOCK_ENABLE, 1);
-       } else {
-               /* DISPCLK */
-               REG_UPDATE_2(DPP_CONTROL[plane_id],
-                       DPPCLK_RATE_CONTROL, 0,
+       else
+               REG_UPDATE(DPP_CONTROL[plane_id],
                        DPP_CLOCK_ENABLE, 1);
-       }
 }
 
 static void enable_power_gating_plane(