#include "intel_dkl_phy.h"
 #include "intel_dkl_phy_regs.h"
 #include "intel_dmc.h"
+#include "intel_dmc_wl.h"
 #include "intel_dp_aux_regs.h"
 #include "intel_dpio_phy.h"
 #include "intel_dpll.h"
                intel_de_rmw(dev_priv, GEN8_CHICKEN_DCPR_1,
                             0, SKL_SELECT_ALTERNATE_DC_EXIT);
 
+       intel_dmc_wl_enable(dev_priv);
+
        gen9_set_dc_state(dev_priv, DC_STATE_EN_UPTO_DC5);
 }
 
                intel_de_rmw(dev_priv, GEN8_CHICKEN_DCPR_1,
                             0, SKL_SELECT_ALTERNATE_DC_EXIT);
 
+       intel_dmc_wl_enable(dev_priv);
+
        gen9_set_dc_state(dev_priv, DC_STATE_EN_UPTO_DC6);
 }
 
        if (!HAS_DISPLAY(dev_priv))
                return;
 
+       intel_dmc_wl_disable(dev_priv);
+
        intel_cdclk_get_cdclk(dev_priv, &cdclk_config);
        /* Can't read out voltage_level so can't use intel_cdclk_changed() */
        drm_WARN_ON(&dev_priv->drm,
 
        pipedmc_clock_gating_wa(i915, true);
        disable_all_event_handlers(i915);
        pipedmc_clock_gating_wa(i915, false);
+
+       intel_dmc_wl_disable(i915);
 }
 
 void assert_dmc_loaded(struct drm_i915_private *i915)
        if (dmc)
                flush_work(&dmc->work);
 
+       intel_dmc_wl_disable(i915);
+
        /* Drop the reference held in case DMC isn't loaded. */
        if (!intel_dmc_has_payload(i915))
                intel_dmc_runtime_pm_put(i915);