1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 0c64fcba..638f47f5 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -4587,6 +4587,12 @@ static int gk20a_init_gr_setup_hw(struct gk20a *g)
         if (err)
                 goto out;
 
+        if (g->ops.gr.init_preemption_state) {
+                err = g->ops.gr.init_preemption_state(g);
+                if (err)
+                        goto out;
+        }
+
         /* disable fe_go_idle */
         gk20a_writel(g, gr_fe_go_idle_timeout_r(),
                 gr_fe_go_idle_timeout_count_disabled_f());