2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 9c09e85f..4f3b34b8 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -414,6 +414,7 @@ struct gpu_ops {
                 u32 (*tpc_enabled_exceptions)(struct gk20a *g);
                 int (*set_czf_bypass)(struct gk20a *g,
                                       struct channel_gk20a *ch);
+                void (*init_czf_bypass)(struct gk20a *g);
                 bool (*sm_debugger_attached)(struct gk20a *g);
                 void (*suspend_single_sm)(struct gk20a *g,
                                 u32 gpc, u32 tpc, u32 sm,
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 6d370250..5910c7d9 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -4838,6 +4838,9 @@ static int gk20a_init_gr_setup_sw(struct gk20a *g)
 
         gr_gk20a_load_zbc_default_table(g, gr);
 
+        if (g->ops.gr.init_czf_bypass)
+                g->ops.gr.init_czf_bypass(g);
+
         nvgpu_mutex_init(&gr->ctx_mutex);
         nvgpu_spinlock_init(&gr->ch_tlb_lock);