/**
  * __set_input_tf - calculates the input transfer function based on expected
  * input space.
+ * @caps: dc color capabilities
  * @func: transfer function
  * @lut: lookup table that defines the color space
  * @lut_size: size of respective lut.
  * Returns:
  * 0 in case of success. -ENOMEM if fails.
  */
-static int __set_input_tf(struct dc_transfer_func *func,
+static int __set_input_tf(struct dc_color_caps *caps, struct dc_transfer_func *func,
                          const struct drm_color_lut *lut, uint32_t lut_size)
 {
        struct dc_gamma *gamma = NULL;
                __drm_lut_to_dc_gamma(lut, gamma, false);
        }
 
-       res = mod_color_calculate_degamma_params(NULL, func, gamma, gamma != NULL);
+       res = mod_color_calculate_degamma_params(caps, func, gamma, gamma != NULL);
 
        if (gamma)
                dc_gamma_release(&gamma);
                func_blend->tf = tf;
                func_blend->sdr_ref_white_level = SDR_WHITE_LEVEL_INIT_VALUE;
 
-               ret = __set_input_tf(func_blend, blend_lut, blend_size);
+               ret = __set_input_tf(NULL, func_blend, blend_lut, blend_size);
        } else {
                func_blend->type = TF_TYPE_BYPASS;
                func_blend->tf = TRANSFER_FUNCTION_LINEAR;
 
 static int
 map_crtc_degamma_to_dc_plane(struct dm_crtc_state *crtc,
-                            struct dc_plane_state *dc_plane_state)
+                            struct dc_plane_state *dc_plane_state,
+                            struct dc_color_caps *caps)
 {
        const struct drm_color_lut *degamma_lut;
        enum dc_transfer_func_predefined tf = TRANSFER_FUNCTION_SRGB;
                        dc_plane_state->in_transfer_func->tf =
                                TRANSFER_FUNCTION_LINEAR;
 
-               r = __set_input_tf(dc_plane_state->in_transfer_func,
+               r = __set_input_tf(caps, dc_plane_state->in_transfer_func,
                                   degamma_lut, degamma_size);
                if (r)
                        return r;
                dc_plane_state->in_transfer_func->tf = tf;
 
                if (tf != TRANSFER_FUNCTION_SRGB &&
-                   !mod_color_calculate_degamma_params(NULL,
+                   !mod_color_calculate_degamma_params(caps,
                                                        dc_plane_state->in_transfer_func,
                                                        NULL, false))
                        return -ENOMEM;
 
 static int
 __set_dm_plane_degamma(struct drm_plane_state *plane_state,
-                      struct dc_plane_state *dc_plane_state)
+                      struct dc_plane_state *dc_plane_state,
+                      struct dc_color_caps *color_caps)
 {
        struct dm_plane_state *dm_plane_state = to_dm_plane_state(plane_state);
        const struct drm_color_lut *degamma_lut;
                dc_plane_state->in_transfer_func->type =
                        TF_TYPE_DISTRIBUTED_POINTS;
 
-               ret = __set_input_tf(dc_plane_state->in_transfer_func,
+               ret = __set_input_tf(color_caps, dc_plane_state->in_transfer_func,
                                     degamma_lut, degamma_size);
                if (ret)
                        return ret;
                dc_plane_state->in_transfer_func->type =
                        TF_TYPE_PREDEFINED;
 
-               if (!mod_color_calculate_degamma_params(NULL,
+               if (!mod_color_calculate_degamma_params(color_caps,
                    dc_plane_state->in_transfer_func, NULL, false))
                        return -ENOMEM;
        }
                                      struct dc_plane_state *dc_plane_state)
 {
        struct amdgpu_device *adev = drm_to_adev(crtc->base.state->dev);
+       struct dc_color_caps *color_caps = NULL;
        bool has_crtc_cm_degamma;
        int ret;
 
                return ret;
        }
 
+       if (dc_plane_state->ctx && dc_plane_state->ctx->dc)
+               color_caps = &dc_plane_state->ctx->dc->caps.color;
+
        /* Initially, we can just bypass the DGM block. */
        dc_plane_state->in_transfer_func->type = TF_TYPE_BYPASS;
        dc_plane_state->in_transfer_func->tf = TRANSFER_FUNCTION_LINEAR;
        /* After, we start to update values according to color props */
        has_crtc_cm_degamma = (crtc->cm_has_degamma || crtc->cm_is_degamma_srgb);
 
-       ret = __set_dm_plane_degamma(plane_state, dc_plane_state);
+       ret = __set_dm_plane_degamma(plane_state, dc_plane_state, color_caps);
        if (ret == -ENOMEM)
                return ret;
 
                 * linearize (implicit degamma) from sRGB/BT709 according to
                 * the input space.
                 */
-               ret = map_crtc_degamma_to_dc_plane(crtc, dc_plane_state);
+               ret = map_crtc_degamma_to_dc_plane(crtc, dc_plane_state, color_caps);
                if (ret)
                        return ret;
        }