3 files changed, 40 insertions, 0 deletions

diff --git a/drivers/gpu/drm/radeon/radeon_display.c b/drivers/gpu/drm/radeon/radeon_display.c
index b92d2f2fcbed..26935cf2c3b3 100644
--- a/drivers/gpu/drm/radeon/radeon_display.c
+++ b/drivers/gpu/drm/radeon/radeon_display.c
@@ -138,6 +138,38 @@ void radeon_crtc_load_lut(struct drm_crtc *crtc)
                 legacy_crtc_load_lut(crtc);
 }
 
+void radeon_crtc_save_lut(struct drm_crtc *crtc)
+{
+        struct radeon_crtc *radeon_crtc = to_radeon_crtc(crtc);
+        int i;
+
+        if (!crtc->enabled)
+                return;
+
+        for (i = 0; i < 256; i++) {
+                radeon_crtc->lut_r_copy[i] = radeon_crtc->lut_r[i];
+                radeon_crtc->lut_g_copy[i] = radeon_crtc->lut_g[i];
+                radeon_crtc->lut_b_copy[i] = radeon_crtc->lut_b[i];
+        }
+}
+
+void radeon_crtc_restore_lut(struct drm_crtc *crtc)
+{
+        struct radeon_crtc *radeon_crtc = to_radeon_crtc(crtc);
+        int i;
+
+        if (!crtc->enabled)
+                return;
+
+        for (i = 0; i < 256; i++) {
+                radeon_crtc->lut_r[i] = radeon_crtc->lut_r_copy[i];
+                radeon_crtc->lut_g[i] = radeon_crtc->lut_g_copy[i];
+                radeon_crtc->lut_b[i] = radeon_crtc->lut_b_copy[i];
+        }
+
+        radeon_crtc_load_lut(crtc);
+}
+
 /** Sets the color ramps on behalf of fbcon */
 void radeon_crtc_fb_gamma_set(struct drm_crtc *crtc, u16 red, u16 green,
                               u16 blue, int regno)
diff --git a/drivers/gpu/drm/radeon/radeon_legacy_crtc.c b/drivers/gpu/drm/radeon/radeon_legacy_crtc.c
index 8752d3447b72..42954785247f 100644
--- a/drivers/gpu/drm/radeon/radeon_legacy_crtc.c
+++ b/drivers/gpu/drm/radeon/radeon_legacy_crtc.c
@@ -355,6 +355,11 @@ int radeon_crtc_set_base_atomic(struct drm_crtc *crtc,
                                 struct drm_framebuffer *fb,
                                 int x, int y, int enter)
 {
+        if (enter)
+                radeon_crtc_save_lut(crtc);
+        else
+                radeon_crtc_restore_lut(crtc);
+
         return radeon_crtc_do_set_base(crtc, fb, x, y, 1);
 }
 
diff --git a/drivers/gpu/drm/radeon/radeon_mode.h b/drivers/gpu/drm/radeon/radeon_mode.h
index c4116d3d8d06..2f78615f02aa 100644
--- a/drivers/gpu/drm/radeon/radeon_mode.h
+++ b/drivers/gpu/drm/radeon/radeon_mode.h
@@ -267,6 +267,7 @@ struct radeon_crtc {
         struct drm_crtc base;
         int crtc_id;
         u16 lut_r[256], lut_g[256], lut_b[256];
+        u16 lut_r_copy[256], lut_g_copy[256], lut_b_copy[256];
         bool enabled;
         bool can_tile;
         uint32_t crtc_offset;
@@ -512,6 +513,8 @@ extern int atombios_get_encoder_mode(struct drm_encoder *encoder);
 extern void radeon_encoder_set_active_device(struct drm_encoder *encoder);
 
 extern void radeon_crtc_load_lut(struct drm_crtc *crtc);
+extern void radeon_crtc_save_lut(struct drm_crtc *crtc);
+extern void radeon_crtc_restore_lut(struct drm_crtc *crtc);
 extern int atombios_crtc_set_base(struct drm_crtc *crtc, int x, int y,
                                    struct drm_framebuffer *old_fb);
 extern int atombios_crtc_set_base_atomic(struct drm_crtc *crtc,