rockchip:hdmi: modify the screen to remove when hdmi remove

diff --git a/drivers/video/rockchip/hdmi/rk_hdmi_task.c b/drivers/video/rockchip/hdmi/rk_hdmi_task.c
index 9674407bea98f48669f5a5b5564ddcea86a62231..26fffd4399106de080331cf55e28f32b94ccb2c7 100755 (executable)
--- a/drivers/video/rockchip/hdmi/rk_hdmi_task.c
+++ b/drivers/video/rockchip/hdmi/rk_hdmi_task.c
@@ -81,8 +81,8 @@ void hdmi_sys_remove(struct hdmi *hdmi)
        INIT_LIST_HEAD(&hdmi->edid.modelist);
        hdmi->display = HDMI_DISABLE;
        if (hdmi->set_vif)
-               hdmi->set_vif(hdmi, hdmi->lcdc->screen1, 0);
-       rk_fb_switch_screen(hdmi->lcdc->screen1, 0, hdmi->lcdc->id);
+               hdmi->set_vif(hdmi, hdmi->lcdc->cur_screen, 0);
+       rk_fb_switch_screen(hdmi->lcdc->cur_screen, 0, hdmi->lcdc->id);
        kobject_uevent_env(&hdmi->ddev->dev->kobj, KOBJ_REMOVE, envp);
 
 #ifdef CONFIG_SWITCH