}
 
 
-bool dc_link_set_psr_enable(const struct dc_link *link, bool enable)
+bool dc_link_set_psr_enable(const struct dc_link *link, bool enable, bool wait)
 {
        struct dc  *core_dc = link->ctx->dc;
        struct dmcu *dmcu = core_dc->res_pool->dmcu;
 
        if (dmcu != NULL && link->psr_enabled)
-               dmcu->funcs->set_psr_enable(dmcu, enable);
+               dmcu->funcs->set_psr_enable(dmcu, enable, wait);
 
        return true;
 }
 
                                sizeof(psr_error_status.raw));
 
                        /* PSR error, disable and re-enable PSR */
-                       dc_link_set_psr_enable(link, false);
-                       dc_link_set_psr_enable(link, true);
+                       dc_link_set_psr_enable(link, false, true);
+                       dc_link_set_psr_enable(link, true, true);
 
                        return true;
                } else if (psr_sink_psr_status.bits.SINK_SELF_REFRESH_STATUS ==
 
 bool dc_link_set_backlight_level(const struct dc_link *dc_link, uint32_t level,
                uint32_t frame_ramp, const struct dc_stream_state *stream);
 
-bool dc_link_set_psr_enable(const struct dc_link *dc_link, bool enable);
+bool dc_link_set_psr_enable(const struct dc_link *dc_link, bool enable, bool wait);
 
 bool dc_link_get_psr_state(const struct dc_link *dc_link, uint32_t *psr_state);
 
 
        REG_UPDATE(DMCU_RAM_ACCESS_CTRL, IRAM_HOST_ACCESS_EN, 0);
 }
 
-static void dce_dmcu_set_psr_enable(struct dmcu *dmcu, bool enable)
+static void dce_dmcu_set_psr_enable(struct dmcu *dmcu, bool enable, bool wait)
 {
        struct dce_dmcu *dmcu_dce = TO_DCE_DMCU(dmcu);
        unsigned int dmcu_max_retry_on_wait_reg_ready = 801;
 
        /* notifyDMCUMsg */
        REG_UPDATE(MASTER_COMM_CNTL_REG, MASTER_COMM_INTERRUPT, 1);
-
-       for (retryCount = 0; retryCount <= 100; retryCount++) {
-               dce_get_dmcu_psr_state(dmcu, &psr_state);
-               if (enable) {
-                       if (psr_state != 0)
-                               break;
-               } else {
-                       if (psr_state == 0)
-                               break;
+       if (wait == true) {
+               for (retryCount = 0; retryCount <= 100; retryCount++) {
+                       dce_get_dmcu_psr_state(dmcu, &psr_state);
+                               if (enable) {
+                                       if (psr_state != 0)
+                                               break;
+                               } else {
+                                       if (psr_state == 0)
+                                               break;
+                               }
+                               dm_delay_in_microseconds(dmcu->ctx, 10);
                }
-               dm_delay_in_microseconds(dmcu->ctx, 10);
        }
 }
 
        REG_UPDATE(DMCU_RAM_ACCESS_CTRL, IRAM_HOST_ACCESS_EN, 0);
 }
 
-static void dcn10_dmcu_set_psr_enable(struct dmcu *dmcu, bool enable)
+static void dcn10_dmcu_set_psr_enable(struct dmcu *dmcu, bool enable, bool wait)
 {
        struct dce_dmcu *dmcu_dce = TO_DCE_DMCU(dmcu);
        unsigned int dmcu_max_retry_on_wait_reg_ready = 801;
         *  Exit PSR may need to wait 1-2 frames to power up. Timeout after at
         *  least a few frames. Should never hit the max retry assert below.
         */
+       if (wait == true) {
        for (retryCount = 0; retryCount <= 1000; retryCount++) {
                dcn10_get_dmcu_psr_state(dmcu, &psr_state);
                if (enable) {
 
        /* assert if max retry hit */
        ASSERT(retryCount <= 1000);
+       }
 }
 
 static void dcn10_dmcu_setup_psr(struct dmcu *dmcu,
 
                        unsigned int start_offset,
                        const char *src,
                        unsigned int bytes);
-       void (*set_psr_enable)(struct dmcu *dmcu, bool enable);
+       void (*set_psr_enable)(struct dmcu *dmcu, bool enable, bool wait);
        void (*setup_psr)(struct dmcu *dmcu,
                        struct dc_link *link,
                        struct psr_context *psr_context);