EDP_PSR_MASK(intel_dp->psr.transcoder);
 }
 
-static void psr_irq_control(struct intel_dp *intel_dp)
+static i915_reg_t psr_ctl_reg(struct drm_i915_private *dev_priv,
+                             enum transcoder cpu_transcoder)
 {
-       struct drm_i915_private *dev_priv = dp_to_i915(intel_dp);
-       i915_reg_t imr_reg;
-       u32 mask;
+       return EDP_PSR_CTL(cpu_transcoder);
+}
+
+static i915_reg_t psr_debug_reg(struct drm_i915_private *dev_priv,
+                               enum transcoder cpu_transcoder)
+{
+       return EDP_PSR_DEBUG(cpu_transcoder);
+}
+
+static i915_reg_t psr_perf_cnt_reg(struct drm_i915_private *dev_priv,
+                                  enum transcoder cpu_transcoder)
+{
+       return EDP_PSR_PERF_CNT(cpu_transcoder);
+}
+
+static i915_reg_t psr_status_reg(struct drm_i915_private *dev_priv,
+                                enum transcoder cpu_transcoder)
+{
+       return EDP_PSR_STATUS(cpu_transcoder);
+}
+
+static i915_reg_t psr_imr_reg(struct drm_i915_private *dev_priv,
+                             enum transcoder cpu_transcoder)
+{
+       if (DISPLAY_VER(dev_priv) >= 12)
+               return TRANS_PSR_IMR(cpu_transcoder);
+       else
+               return EDP_PSR_IMR;
+}
 
+static i915_reg_t psr_iir_reg(struct drm_i915_private *dev_priv,
+                             enum transcoder cpu_transcoder)
+{
        if (DISPLAY_VER(dev_priv) >= 12)
-               imr_reg = TRANS_PSR_IMR(intel_dp->psr.transcoder);
+               return TRANS_PSR_IIR(cpu_transcoder);
        else
-               imr_reg = EDP_PSR_IMR;
+               return EDP_PSR_IIR;
+}
+
+static void psr_irq_control(struct intel_dp *intel_dp)
+{
+       struct drm_i915_private *dev_priv = dp_to_i915(intel_dp);
+       enum transcoder cpu_transcoder = intel_dp->psr.transcoder;
+       u32 mask;
 
        mask = psr_irq_psr_error_bit_get(intel_dp);
        if (intel_dp->psr.debug & I915_PSR_DEBUG_IRQ)
                mask |= psr_irq_post_exit_bit_get(intel_dp) |
                        psr_irq_pre_entry_bit_get(intel_dp);
 
-       intel_de_rmw(dev_priv, imr_reg, psr_irq_mask_get(intel_dp), ~mask);
+       intel_de_rmw(dev_priv, psr_imr_reg(dev_priv, cpu_transcoder),
+                    psr_irq_mask_get(intel_dp), ~mask);
 }
 
 static void psr_event_print(struct drm_i915_private *i915,
        struct drm_i915_private *dev_priv = dp_to_i915(intel_dp);
        enum transcoder cpu_transcoder = intel_dp->psr.transcoder;
        ktime_t time_ns =  ktime_get();
-       i915_reg_t imr_reg;
-
-       if (DISPLAY_VER(dev_priv) >= 12)
-               imr_reg = TRANS_PSR_IMR(cpu_transcoder);
-       else
-               imr_reg = EDP_PSR_IMR;
 
        if (psr_iir & psr_irq_pre_entry_bit_get(intel_dp)) {
                intel_dp->psr.last_entry_attempt = time_ns;
                 * again so we don't care about unmask the interruption
                 * or unset irq_aux_error.
                 */
-               intel_de_rmw(dev_priv, imr_reg, 0, psr_irq_psr_error_bit_get(intel_dp));
+               intel_de_rmw(dev_priv, psr_imr_reg(dev_priv, cpu_transcoder),
+                            0, psr_irq_psr_error_bit_get(intel_dp));
 
                queue_work(dev_priv->unordered_wq, &intel_dp->psr.work);
        }
        if (DISPLAY_VER(dev_priv) >= 8)
                val |= EDP_PSR_CRC_ENABLE;
 
-       intel_de_rmw(dev_priv, EDP_PSR_CTL(cpu_transcoder),
+       intel_de_rmw(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder),
                     ~EDP_PSR_RESTORE_PSR_ACTIVE_CTX_MASK, val);
 }
 
         * PSR2 HW is incorrectly using EDP_PSR_TP1_TP3_SEL and BSpec is
         * recommending keep this bit unset while PSR2 is enabled.
         */
-       intel_de_write(dev_priv, EDP_PSR_CTL(cpu_transcoder), 0);
+       intel_de_write(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder), 0);
 
        intel_de_write(dev_priv, EDP_PSR2_CTL(cpu_transcoder), val);
 }
        struct drm_i915_private *dev_priv = dp_to_i915(intel_dp);
        enum transcoder cpu_transcoder = intel_dp->psr.transcoder;
 
-       if (transcoder_has_psr2(dev_priv, cpu_transcoder))
-               drm_WARN_ON(&dev_priv->drm,
-                           intel_de_read(dev_priv, EDP_PSR2_CTL(cpu_transcoder)) & EDP_PSR2_ENABLE);
+       drm_WARN_ON(&dev_priv->drm,
+                   transcoder_has_psr2(dev_priv, cpu_transcoder) &&
+                   intel_de_read(dev_priv, EDP_PSR2_CTL(cpu_transcoder)) & EDP_PSR2_ENABLE);
 
        drm_WARN_ON(&dev_priv->drm,
-                   intel_de_read(dev_priv, EDP_PSR_CTL(cpu_transcoder)) & EDP_PSR_ENABLE);
+                   intel_de_read(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder)) & EDP_PSR_ENABLE);
+
        drm_WARN_ON(&dev_priv->drm, intel_dp->psr.active);
+
        lockdep_assert_held(&intel_dp->psr.lock);
 
        /* psr1 and psr2 are mutually exclusive.*/
        if (DISPLAY_VER(dev_priv) < 11)
                mask |= EDP_PSR_DEBUG_MASK_DISP_REG_WRITE;
 
-       intel_de_write(dev_priv, EDP_PSR_DEBUG(cpu_transcoder),
-                      mask);
+       intel_de_write(dev_priv, psr_debug_reg(dev_priv, cpu_transcoder), mask);
 
        psr_irq_control(intel_dp);
 
         * first time that PSR HW tries to activate so lets keep PSR disabled
         * to avoid any rendering problems.
         */
-       if (DISPLAY_VER(dev_priv) >= 12)
-               val = intel_de_read(dev_priv, TRANS_PSR_IIR(cpu_transcoder));
-       else
-               val = intel_de_read(dev_priv, EDP_PSR_IIR);
+       val = intel_de_read(dev_priv, psr_iir_reg(dev_priv, cpu_transcoder));
        val &= psr_irq_psr_error_bit_get(intel_dp);
        if (val) {
                intel_dp->psr.sink_not_reliable = true;
                        drm_WARN_ON(&dev_priv->drm, val & EDP_PSR2_ENABLE);
                }
 
-               val = intel_de_read(dev_priv, EDP_PSR_CTL(cpu_transcoder));
+               val = intel_de_read(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder));
                drm_WARN_ON(&dev_priv->drm, val & EDP_PSR_ENABLE);
 
                return;
 
                drm_WARN_ON(&dev_priv->drm, !(val & EDP_PSR2_ENABLE));
        } else {
-               val = intel_de_rmw(dev_priv, EDP_PSR_CTL(cpu_transcoder),
+               val = intel_de_rmw(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder),
                                   EDP_PSR_ENABLE, 0);
 
                drm_WARN_ON(&dev_priv->drm, !(val & EDP_PSR_ENABLE));
                psr_status = EDP_PSR2_STATUS(cpu_transcoder);
                psr_status_mask = EDP_PSR2_STATUS_STATE_MASK;
        } else {
-               psr_status = EDP_PSR_STATUS(cpu_transcoder);
+               psr_status = psr_status_reg(dev_priv, cpu_transcoder);
                psr_status_mask = EDP_PSR_STATUS_STATE_MASK;
        }
 
         * defensive enough to cover everything.
         */
        return intel_de_wait_for_clear(dev_priv,
-                                      EDP_PSR_STATUS(cpu_transcoder),
+                                      psr_status_reg(dev_priv, cpu_transcoder),
                                       EDP_PSR_STATUS_STATE_MASK, 50);
 }
 
                reg = EDP_PSR2_STATUS(cpu_transcoder);
                mask = EDP_PSR2_STATUS_STATE_MASK;
        } else {
-               reg = EDP_PSR_STATUS(cpu_transcoder);
+               reg = psr_status_reg(dev_priv, cpu_transcoder);
                mask = EDP_PSR_STATUS_STATE_MASK;
        }
 
                        "SRDOFFACK",
                        "SRDENT_ON",
                };
-               val = intel_de_read(dev_priv, EDP_PSR_STATUS(cpu_transcoder));
+               val = intel_de_read(dev_priv, psr_status_reg(dev_priv, cpu_transcoder));
                status_val = REG_FIELD_GET(EDP_PSR_STATUS_STATE_MASK, val);
                if (status_val < ARRAY_SIZE(live_status))
                        status = live_status[status_val];
                val = intel_de_read(dev_priv, EDP_PSR2_CTL(cpu_transcoder));
                enabled = val & EDP_PSR2_ENABLE;
        } else {
-               val = intel_de_read(dev_priv, EDP_PSR_CTL(cpu_transcoder));
+               val = intel_de_read(dev_priv, psr_ctl_reg(dev_priv, cpu_transcoder));
                enabled = val & EDP_PSR_ENABLE;
        }
        seq_printf(m, "Source PSR ctl: %s [0x%08x]\n",
        /*
         * SKL+ Perf counter is reset to 0 everytime DC state is entered
         */
-       val = intel_de_read(dev_priv, EDP_PSR_PERF_CNT(cpu_transcoder));
+       val = intel_de_read(dev_priv, psr_perf_cnt_reg(dev_priv, cpu_transcoder));
        seq_printf(m, "Performance counter: %u\n",
                   REG_FIELD_GET(EDP_PSR_PERF_CNT_MASK, val));