1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 638f47f5..4a3d58cb 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -1571,6 +1571,9 @@ static int gr_gk20a_init_golden_ctx_image(struct gk20a *g,
                 gk20a_writel(g, sw_ctx_load->l[i].addr,
                              sw_ctx_load->l[i].value);
 
+        if (g->ops.gr.init_preemption_state)
+                g->ops.gr.init_preemption_state(g);
+
         g->ops.clock_gating.blcg_gr_load_gating_prod(g, g->blcg_enabled);
 
         err = gr_gk20a_wait_idle(g, end_jiffies, GR_IDLE_CHECK_DEFAULT);