const struct intel_crtc_state *crtc_state)
 {
        /* Cannot enable DSC and PSR2 simultaneously */
-       WARN_ON(crtc_state->dsc.compression_enable &&
-               crtc_state->has_psr2);
+       drm_WARN_ON(&dev_priv->drm, crtc_state->dsc.compression_enable &&
+                   crtc_state->has_psr2);
 
        switch (dev_priv->psr.debug & I915_PSR_DEBUG_MODE_MASK) {
        case I915_PSR_DEBUG_DISABLE:
        idle_frames = max(6, dev_priv->vbt.psr.idle_frames);
        idle_frames = max(idle_frames, dev_priv->psr.sink_sync_latency + 1);
 
-       if (WARN_ON(idle_frames > 0xf))
+       if (drm_WARN_ON(&dev_priv->drm, idle_frames > 0xf))
                idle_frames = 0xf;
 
        return idle_frames;
        exit_scanlines =
                intel_usecs_to_scanlines(&crtc_state->uapi.adjusted_mode, 200) + 1;
 
-       if (WARN_ON(exit_scanlines > crtc_vdisplay))
+       if (drm_WARN_ON(&dev_priv->drm, exit_scanlines > crtc_vdisplay))
                return;
 
        crtc_state->dc3co_exitline = crtc_vdisplay - exit_scanlines;
        struct drm_i915_private *dev_priv = dp_to_i915(intel_dp);
 
        if (transcoder_has_psr2(dev_priv, dev_priv->psr.transcoder))
-               WARN_ON(intel_de_read(dev_priv, EDP_PSR2_CTL(dev_priv->psr.transcoder)) & EDP_PSR2_ENABLE);
+               drm_WARN_ON(&dev_priv->drm,
+                           intel_de_read(dev_priv, EDP_PSR2_CTL(dev_priv->psr.transcoder)) & EDP_PSR2_ENABLE);
 
-       WARN_ON(intel_de_read(dev_priv, EDP_PSR_CTL(dev_priv->psr.transcoder)) & EDP_PSR_ENABLE);
-       WARN_ON(dev_priv->psr.active);
+       drm_WARN_ON(&dev_priv->drm,
+                   intel_de_read(dev_priv, EDP_PSR_CTL(dev_priv->psr.transcoder)) & EDP_PSR_ENABLE);
+       drm_WARN_ON(&dev_priv->drm, dev_priv->psr.active);
        lockdep_assert_held(&dev_priv->psr.lock);
 
        /* psr1 and psr2 are mutually exclusive.*/
        struct intel_dp *intel_dp = dev_priv->psr.dp;
        u32 val;
 
-       WARN_ON(dev_priv->psr.enabled);
+       drm_WARN_ON(&dev_priv->drm, dev_priv->psr.enabled);
 
        dev_priv->psr.psr2_enabled = intel_psr2_enabled(dev_priv, crtc_state);
        dev_priv->psr.busy_frontbuffer_bits = 0;
        if (!crtc_state->has_psr)
                return;
 
-       if (WARN_ON(!CAN_PSR(dev_priv)))
+       if (drm_WARN_ON(&dev_priv->drm, !CAN_PSR(dev_priv)))
                return;
 
-       WARN_ON(dev_priv->drrs.dp);
+       drm_WARN_ON(&dev_priv->drm, dev_priv->drrs.dp);
 
        mutex_lock(&dev_priv->psr.lock);
 
                if (transcoder_has_psr2(dev_priv, dev_priv->psr.transcoder)) {
                        val = intel_de_read(dev_priv,
                                            EDP_PSR2_CTL(dev_priv->psr.transcoder));
-                       WARN_ON(val & EDP_PSR2_ENABLE);
+                       drm_WARN_ON(&dev_priv->drm, val & EDP_PSR2_ENABLE);
                }
 
                val = intel_de_read(dev_priv,
                                    EDP_PSR_CTL(dev_priv->psr.transcoder));
-               WARN_ON(val & EDP_PSR_ENABLE);
+               drm_WARN_ON(&dev_priv->drm, val & EDP_PSR_ENABLE);
 
                return;
        }
                tgl_disallow_dc3co_on_psr2_exit(dev_priv);
                val = intel_de_read(dev_priv,
                                    EDP_PSR2_CTL(dev_priv->psr.transcoder));
-               WARN_ON(!(val & EDP_PSR2_ENABLE));
+               drm_WARN_ON(&dev_priv->drm, !(val & EDP_PSR2_ENABLE));
                val &= ~EDP_PSR2_ENABLE;
                intel_de_write(dev_priv,
                               EDP_PSR2_CTL(dev_priv->psr.transcoder), val);
        } else {
                val = intel_de_read(dev_priv,
                                    EDP_PSR_CTL(dev_priv->psr.transcoder));
-               WARN_ON(!(val & EDP_PSR_ENABLE));
+               drm_WARN_ON(&dev_priv->drm, !(val & EDP_PSR_ENABLE));
                val &= ~EDP_PSR_ENABLE;
                intel_de_write(dev_priv,
                               EDP_PSR_CTL(dev_priv->psr.transcoder), val);
        if (!old_crtc_state->has_psr)
                return;
 
-       if (WARN_ON(!CAN_PSR(dev_priv)))
+       if (drm_WARN_ON(&dev_priv->drm, !CAN_PSR(dev_priv)))
                return;
 
        mutex_lock(&dev_priv->psr.lock);