dp_info->tries = 0;
        channel_eq = false;
        while (1) {
-               drm_dp_link_train_channel_eq_delay(dp_info->dpcd);
+               drm_dp_link_train_channel_eq_delay(dp_info->aux, dp_info->dpcd);
 
                if (drm_dp_dpcd_read_link_status(dp_info->aux,
                                                 dp_info->link_status) <= 0) {
 
 }
 EXPORT_SYMBOL(drm_dp_link_train_clock_recovery_delay);
 
-static void __drm_dp_link_train_channel_eq_delay(unsigned long rd_interval)
+static void __drm_dp_link_train_channel_eq_delay(const struct drm_dp_aux *aux,
+                                                unsigned long rd_interval)
 {
        if (rd_interval > 4)
                DRM_DEBUG_KMS("AUX interval %lu, out of range (max 4)\n",
        usleep_range(rd_interval, rd_interval * 2);
 }
 
-void drm_dp_link_train_channel_eq_delay(const u8 dpcd[DP_RECEIVER_CAP_SIZE])
+void drm_dp_link_train_channel_eq_delay(const struct drm_dp_aux *aux,
+                                       const u8 dpcd[DP_RECEIVER_CAP_SIZE])
 {
-       __drm_dp_link_train_channel_eq_delay(dpcd[DP_TRAINING_AUX_RD_INTERVAL] &
+       __drm_dp_link_train_channel_eq_delay(aux,
+                                            dpcd[DP_TRAINING_AUX_RD_INTERVAL] &
                                             DP_TRAINING_AUX_RD_MASK);
 }
 EXPORT_SYMBOL(drm_dp_link_train_channel_eq_delay);
        return phy_cap[r - DP_TRAINING_AUX_RD_INTERVAL_PHY_REPEATER1];
 }
 
-void drm_dp_lttpr_link_train_channel_eq_delay(const u8 phy_cap[DP_LTTPR_PHY_CAP_SIZE])
+void drm_dp_lttpr_link_train_channel_eq_delay(const struct drm_dp_aux *aux,
+                                             const u8 phy_cap[DP_LTTPR_PHY_CAP_SIZE])
 {
        u8 interval = dp_lttpr_phy_cap(phy_cap,
                                       DP_TRAINING_AUX_RD_INTERVAL_PHY_REPEATER1) &
                      DP_TRAINING_AUX_RD_MASK;
 
-       __drm_dp_link_train_channel_eq_delay(interval);
+       __drm_dp_link_train_channel_eq_delay(aux, interval);
 }
 EXPORT_SYMBOL(drm_dp_lttpr_link_train_channel_eq_delay);
 
 
                                                  enum drm_dp_phy dp_phy)
 {
        if (dp_phy == DP_PHY_DPRX) {
-               drm_dp_link_train_channel_eq_delay(intel_dp->dpcd);
+               drm_dp_link_train_channel_eq_delay(&intel_dp->aux, intel_dp->dpcd);
        } else {
                const u8 *phy_caps = intel_dp_lttpr_phy_caps(intel_dp, dp_phy);
 
-               drm_dp_lttpr_link_train_channel_eq_delay(phy_caps);
+               drm_dp_lttpr_link_train_channel_eq_delay(&intel_dp->aux, phy_caps);
        }
 }
 
 
 static void dp_ctrl_clear_training_pattern(struct dp_ctrl_private *ctrl)
 {
        dp_ctrl_train_pattern_set(ctrl, DP_TRAINING_PATTERN_DISABLE);
-       drm_dp_link_train_channel_eq_delay(ctrl->panel->dpcd);
+       drm_dp_link_train_channel_eq_delay(ctrl->aux, ctrl->panel->dpcd);
 }
 
 static int dp_ctrl_link_train_2(struct dp_ctrl_private *ctrl,
        dp_ctrl_train_pattern_set(ctrl, pattern | DP_RECOVERED_CLOCK_OUT_EN);
 
        for (tries = 0; tries <= maximum_retries; tries++) {
-               drm_dp_link_train_channel_eq_delay(ctrl->panel->dpcd);
+               drm_dp_link_train_channel_eq_delay(ctrl->aux, ctrl->panel->dpcd);
 
                ret = dp_ctrl_read_link_status(ctrl, link_status);
                if (ret)
 
                return ret;
 
        while (1) {
-               drm_dp_link_train_channel_eq_delay(ctrl->dpcd);
+               drm_dp_link_train_channel_eq_delay(ctrl->drm_aux, ctrl->dpcd);
 
                rlen = drm_dp_dpcd_read_link_status(ctrl->drm_aux, link_status);
                if (rlen < DP_LINK_STATUS_SIZE) {
 
        ret = edp_train_pattern_set_write(ctrl, 0);
 
-       drm_dp_link_train_channel_eq_delay(ctrl->dpcd);
+       drm_dp_link_train_channel_eq_delay(ctrl->drm_aux, ctrl->dpcd);
 
        return ret;
 }
 
        dp_info->tries = 0;
        channel_eq = false;
        while (1) {
-               drm_dp_link_train_channel_eq_delay(dp_info->dpcd);
+               drm_dp_link_train_channel_eq_delay(dp_info->aux, dp_info->dpcd);
 
                if (drm_dp_dpcd_read_link_status(dp_info->aux,
                                                 dp_info->link_status) <= 0) {
 
                if (ret)
                        return ret;
 
-               drm_dp_link_train_channel_eq_delay(dp->dpcd);
+               drm_dp_link_train_channel_eq_delay(&dp->aux, dp->dpcd);
                ret = drm_dp_dpcd_read_link_status(&dp->aux, link_status);
                if (ret < 0)
                        return ret;
 
 void drm_dp_link_train_clock_recovery_delay(const struct drm_dp_aux *aux,
                                            const u8 dpcd[DP_RECEIVER_CAP_SIZE]);
 void drm_dp_lttpr_link_train_clock_recovery_delay(void);
-void drm_dp_link_train_channel_eq_delay(const u8 dpcd[DP_RECEIVER_CAP_SIZE]);
-void drm_dp_lttpr_link_train_channel_eq_delay(const u8 caps[DP_LTTPR_PHY_CAP_SIZE]);
+void drm_dp_link_train_channel_eq_delay(const struct drm_dp_aux *aux,
+                                       const u8 dpcd[DP_RECEIVER_CAP_SIZE]);
+void drm_dp_lttpr_link_train_channel_eq_delay(const struct drm_dp_aux *aux,
+                                             const u8 caps[DP_LTTPR_PHY_CAP_SIZE]);
 
 u8 drm_dp_link_rate_to_bw_code(int link_rate);
 int drm_dp_bw_code_to_link_rate(u8 link_bw);