From: Yongqiang Sun <[email protected]>

Change-Id: Iff63270cf76bcd0cb8f76342df3d333425c89b85
Signed-off-by: Yongqiang Sun <[email protected]>
Reviewed-by: Tony Cheng <[email protected]>
Acked-by: Harry Wentland <[email protected]>
---
 drivers/gpu/drm/amd/display/dc/core/dc.c          | 2 +-
 drivers/gpu/drm/amd/display/dc/core/dc_surface.c  | 9 +++++----
 drivers/gpu/drm/amd/display/dc/dc.h               | 2 +-
 drivers/gpu/drm/amd/display/modules/color/color.c | 8 +++-----
 4 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/drivers/gpu/drm/amd/display/dc/core/dc.c 
b/drivers/gpu/drm/amd/display/dc/core/dc.c
index aca13d1cd4bb..12620df7056d 100644
--- a/drivers/gpu/drm/amd/display/dc/core/dc.c
+++ b/drivers/gpu/drm/amd/display/dc/core/dc.c
@@ -1458,7 +1458,7 @@ void dc_update_surfaces_for_target(struct dc *dc, struct 
dc_surface_update *upda
                        if (updates[i].gamma &&
                                updates[i].gamma != 
surface->public.gamma_correction) {
                                if (surface->public.gamma_correction != NULL)
-                                       dc_gamma_release(surface->public.
+                                       dc_gamma_release(&surface->public.
                                                        gamma_correction);
 
                                dc_gamma_retain(updates[i].gamma);
diff --git a/drivers/gpu/drm/amd/display/dc/core/dc_surface.c 
b/drivers/gpu/drm/amd/display/dc/core/dc_surface.c
index d962baa477f5..d607972b830b 100644
--- a/drivers/gpu/drm/amd/display/dc/core/dc_surface.c
+++ b/drivers/gpu/drm/amd/display/dc/core/dc_surface.c
@@ -74,8 +74,7 @@ static bool construct(struct dc_context *ctx, struct surface 
*surface)
 static void destruct(struct surface *surface)
 {
        if (surface->protected.public.gamma_correction != NULL) {
-               dc_gamma_release(surface->protected.public.gamma_correction);
-               surface->protected.public.gamma_correction = NULL;
+               dc_gamma_release(&surface->protected.public.gamma_correction);
        }
        if (surface->protected.public.in_transfer_func != NULL) {
                dc_transfer_func_release(
@@ -189,15 +188,17 @@ void dc_gamma_retain(const struct dc_gamma *dc_gamma)
        ++gamma->ref_count;
 }
 
-void dc_gamma_release(const struct dc_gamma *dc_gamma)
+void dc_gamma_release(const struct dc_gamma **dc_gamma)
 {
-       struct gamma *gamma = DC_GAMMA_TO_GAMMA(dc_gamma);
+       struct gamma *gamma = DC_GAMMA_TO_GAMMA(*dc_gamma);
 
        ASSERT(gamma->ref_count > 0);
        --gamma->ref_count;
 
        if (gamma->ref_count == 0)
                dm_free(gamma);
+
+       *dc_gamma = NULL;
 }
 
 struct dc_gamma *dc_create_gamma()
diff --git a/drivers/gpu/drm/amd/display/dc/dc.h 
b/drivers/gpu/drm/amd/display/dc/dc.h
index 23f41b690c02..599f8b0894c2 100644
--- a/drivers/gpu/drm/amd/display/dc/dc.h
+++ b/drivers/gpu/drm/amd/display/dc/dc.h
@@ -318,7 +318,7 @@ void dc_surface_retain(const struct dc_surface *dc_surface);
 void dc_surface_release(const struct dc_surface *dc_surface);
 
 void dc_gamma_retain(const struct dc_gamma *dc_gamma);
-void dc_gamma_release(const struct dc_gamma *dc_gamma);
+void dc_gamma_release(const struct dc_gamma **dc_gamma);
 struct dc_gamma *dc_create_gamma(void);
 
 void dc_transfer_func_retain(const struct dc_transfer_func *dc_tf);
diff --git a/drivers/gpu/drm/amd/display/modules/color/color.c 
b/drivers/gpu/drm/amd/display/modules/color/color.c
index 599d9f9aa7ca..3611de90e1e9 100644
--- a/drivers/gpu/drm/amd/display/modules/color/color.c
+++ b/drivers/gpu/drm/amd/display/modules/color/color.c
@@ -1557,7 +1557,7 @@ void mod_color_destroy(struct mod_color *mod_color)
 
                for (i = 0; i < core_color->num_sinks; i++)
                        if (core_color->state[i].gamma)
-                               dc_gamma_release(core_color->state[i].gamma);
+                               dc_gamma_release(&core_color->state[i].gamma);
 
                dm_free(core_color->state);
 
@@ -1749,8 +1749,7 @@ bool mod_color_remove_sink(struct mod_color *mod_color,
        for (i = 0; i < core_color->num_sinks; i++) {
                if (core_color->caps[i].sink == sink) {
                        if (core_color->state[i].gamma) {
-                               dc_gamma_release(core_color->state[i].gamma);
-                               core_color->state[i].gamma = NULL;
+                               dc_gamma_release(&core_color->state[i].gamma);
                        }
 
                        /* To remove this sink, shift everything after down */
@@ -2444,8 +2443,7 @@ bool mod_color_set_input_gamma_correction(struct 
mod_color *mod_color,
                        if (core_color->state[sink_index].gamma != gamma) {
                                if (core_color->state[sink_index].gamma)
                                        dc_gamma_release(
-                                               core_color->state[sink_index].
-                                               gamma);
+                                               
&core_color->state[sink_index].gamma);
 
                                dc_gamma_retain(gamma);
                                core_color->state[sink_index].gamma = gamma;
-- 
2.9.3

_______________________________________________
amd-gfx mailing list
[email protected]
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to