1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/cde_gk20a.c b/drivers/gpu/nvgpu/gk20a/cde_gk20a.c
index ddca39f3..cd4e71bf 100644
--- a/drivers/gpu/nvgpu/gk20a/cde_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/cde_gk20a.c
@@ -493,6 +493,9 @@ static int gk20a_init_cde_required_class(struct gk20a_cde_ctx *cde_ctx,
         alloc_obj_ctx.class_num = required_class;
         alloc_obj_ctx.flags = 0;
 
+        /* CDE enabled */
+        cde_ctx->ch->cde = true;
+
         err = gk20a_alloc_obj_ctx(cde_ctx->ch, &alloc_obj_ctx);
         if (err) {
                 gk20a_warn(&cde_ctx->pdev->dev, "cde: failed to allocate ctx. err=%d",