2 files changed, 5 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
index 6ee73ffe..396c5ee5 100644
--- a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
@@ -379,8 +379,7 @@ int gk20a_init_fifo_reset_enable_hw(struct gk20a *g)
 
         gk20a_dbg_fn("");
         /* enable pmc pfifo */
-        gk20a_reset(g, mc_enable_pfifo_enabled_f()
-                        | mc_enable_ce2_enabled_f());
+        gk20a_reset(g, mc_enable_pfifo_enabled_f());
 
         if (g->ops.clock_gating.slcg_ce2_load_gating_prod)
                 g->ops.clock_gating.slcg_ce2_load_gating_prod(g,
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 943b4085..60247da8 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -4550,6 +4550,8 @@ static int gk20a_init_gr_prepare(struct gk20a *g)
 {
         u32 gpfifo_ctrl, pmc_en;
         u32 err = 0;
+        struct fifo_engine_info_gk20a *ce_info =
+                g->fifo.engine_info + ENGINE_CE2_GK20A;
 
         /* disable fifo access */
         pmc_en = gk20a_readl(g, mc_enable_r());
@@ -4562,7 +4564,8 @@ static int gk20a_init_gr_prepare(struct gk20a *g)
         /* reset gr engine */
         gk20a_reset(g, mc_enable_pgraph_enabled_f()
                         | mc_enable_blg_enabled_f()
-                        | mc_enable_perfmon_enabled_f());
+                        | mc_enable_perfmon_enabled_f()
+                        | ce_info->reset_mask);
 
         gr_gk20a_load_gating_prod(g);
         /* Disable elcg until it gets enabled later in the init*/