/*
  * struct dpu_plane - local dpu plane structure
  * @aspace: address space pointer
- * @csc_ptr: Points to dpu_csc_cfg structure to use for current
  * @mplane_list: List of multirect planes of the same pipe
  * @catalog: Points to dpu catalog structure
  * @revalidate: force revalidation of all the plane properties
        struct list_head mplane_list;
        struct dpu_mdss_cfg *catalog;
 
-       struct dpu_csc_cfg *csc_ptr;
-
        const struct dpu_sspp_sub_blks *pipe_sblk;
 
        /* debugfs related stuff */
        scale_cfg->enable = 1;
 }
 
-static void _dpu_plane_setup_csc(struct dpu_plane *pdpu)
-{
-       static const struct dpu_csc_cfg dpu_csc_YUV2RGB_601L = {
-               {
-                       /* S15.16 format */
-                       0x00012A00, 0x00000000, 0x00019880,
-                       0x00012A00, 0xFFFF9B80, 0xFFFF3000,
-                       0x00012A00, 0x00020480, 0x00000000,
+static const struct dpu_csc_cfg dpu_csc_YUV2RGB_601L = {
+       {
+               /* S15.16 format */
+               0x00012A00, 0x00000000, 0x00019880,
+               0x00012A00, 0xFFFF9B80, 0xFFFF3000,
+               0x00012A00, 0x00020480, 0x00000000,
+       },
+       /* signed bias */
+       { 0xfff0, 0xff80, 0xff80,},
+       { 0x0, 0x0, 0x0,},
+       /* unsigned clamp */
+       { 0x10, 0xeb, 0x10, 0xf0, 0x10, 0xf0,},
+       { 0x00, 0xff, 0x00, 0xff, 0x00, 0xff,},
+};
+
+static const struct dpu_csc_cfg dpu_csc10_YUV2RGB_601L = {
+       {
+               /* S15.16 format */
+               0x00012A00, 0x00000000, 0x00019880,
+               0x00012A00, 0xFFFF9B80, 0xFFFF3000,
+               0x00012A00, 0x00020480, 0x00000000,
                },
-               /* signed bias */
-               { 0xfff0, 0xff80, 0xff80,},
-               { 0x0, 0x0, 0x0,},
-               /* unsigned clamp */
-               { 0x10, 0xeb, 0x10, 0xf0, 0x10, 0xf0,},
-               { 0x00, 0xff, 0x00, 0xff, 0x00, 0xff,},
-       };
-       static const struct dpu_csc_cfg dpu_csc10_YUV2RGB_601L = {
-               {
-                       /* S15.16 format */
-                       0x00012A00, 0x00000000, 0x00019880,
-                       0x00012A00, 0xFFFF9B80, 0xFFFF3000,
-                       0x00012A00, 0x00020480, 0x00000000,
-                       },
-               /* signed bias */
-               { 0xffc0, 0xfe00, 0xfe00,},
-               { 0x0, 0x0, 0x0,},
-               /* unsigned clamp */
-               { 0x40, 0x3ac, 0x40, 0x3c0, 0x40, 0x3c0,},
-               { 0x00, 0x3ff, 0x00, 0x3ff, 0x00, 0x3ff,},
-       };
+       /* signed bias */
+       { 0xffc0, 0xfe00, 0xfe00,},
+       { 0x0, 0x0, 0x0,},
+       /* unsigned clamp */
+       { 0x40, 0x3ac, 0x40, 0x3c0, 0x40, 0x3c0,},
+       { 0x00, 0x3ff, 0x00, 0x3ff, 0x00, 0x3ff,},
+};
+
+static const struct dpu_csc_cfg *_dpu_plane_get_csc(struct dpu_plane *pdpu, const struct dpu_format *fmt)
+{
+       const struct dpu_csc_cfg *csc_ptr;
 
        if (!pdpu) {
                DPU_ERROR("invalid plane\n");
-               return;
+               return NULL;
        }
 
+       if (!DPU_FORMAT_IS_YUV(fmt))
+               return NULL;
+
        if (BIT(DPU_SSPP_CSC_10BIT) & pdpu->features)
-               pdpu->csc_ptr = (struct dpu_csc_cfg *)&dpu_csc10_YUV2RGB_601L;
+               csc_ptr = &dpu_csc10_YUV2RGB_601L;
        else
-               pdpu->csc_ptr = (struct dpu_csc_cfg *)&dpu_csc_YUV2RGB_601L;
+               csc_ptr = &dpu_csc_YUV2RGB_601L;
 
        DPU_DEBUG_PLANE(pdpu, "using 0x%X 0x%X 0x%X...\n",
-                       pdpu->csc_ptr->csc_mv[0],
-                       pdpu->csc_ptr->csc_mv[1],
-                       pdpu->csc_ptr->csc_mv[2]);
+                       csc_ptr->csc_mv[0],
+                       csc_ptr->csc_mv[1],
+                       csc_ptr->csc_mv[2]);
+
+       return csc_ptr;
 }
 
 static void _dpu_plane_setup_scaler(struct dpu_plane *pdpu,
        else if (pdpu->color_fill & DPU_PLANE_COLOR_FILL_FLAG)
                /* force 100% alpha */
                _dpu_plane_color_fill(pdpu, pdpu->color_fill, 0xFF);
-       else if (pdpu->pipe_hw && pdpu->csc_ptr && pdpu->pipe_hw->ops.setup_csc)
-               pdpu->pipe_hw->ops.setup_csc(pdpu->pipe_hw, pdpu->csc_ptr);
+       else if (pdpu->pipe_hw && pdpu->pipe_hw->ops.setup_csc) {
+               const struct dpu_format *fmt = to_dpu_format(msm_framebuffer_format(plane->state->fb));
+               const struct dpu_csc_cfg *csc_ptr = _dpu_plane_get_csc(pdpu, fmt);
+
+               if (csc_ptr)
+                       pdpu->pipe_hw->ops.setup_csc(pdpu->pipe_hw, csc_ptr);
+       }
 
        /* flag h/w flush complete */
        if (plane->state)
 
                        pdpu->pipe_hw->ops.setup_cdp(pdpu->pipe_hw, cdp_cfg);
                }
-
-               /* update csc */
-               if (DPU_FORMAT_IS_YUV(fmt))
-                       _dpu_plane_setup_csc(pdpu);
-               else
-                       pdpu->csc_ptr = NULL;
        }
 
        _dpu_plane_set_qos_lut(plane, fb, &pipe_cfg);