2 files changed, 1 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
index ee49a8f2..50b819ac 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
@@ -2309,10 +2309,6 @@ void gv11b_restore_context_header(struct gk20a *g,
                         ctxsw_prog_main_image_context_buffer_ptr_hi_o(), va_hi);
         nvgpu_mem_wr(g, ctxheader,
                         ctxsw_prog_main_image_context_buffer_ptr_o(), va_lo);
-        nvgpu_mem_wr(g, ctxheader,
-                        ctxsw_prog_main_image_num_restore_ops_o(), 0);
-        nvgpu_mem_wr(g, ctxheader,
-                        ctxsw_prog_main_image_num_save_ops_o(), 0);
 }
 void gr_gv11b_write_zcull_ptr(struct gk20a *g,
                                 struct nvgpu_mem *mem, u64 gpu_va)
diff --git a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
index 2abd6243..be3f2456 100644
--- a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
@@ -343,6 +343,7 @@ static const struct gpu_ops gv11b_ops = {
                 .handle_tpc_sm_ecc_exception =
                         gr_gv11b_handle_tpc_sm_ecc_exception,
                 .decode_egpc_addr = gv11b_gr_decode_egpc_addr,
+                .init_ctxsw_hdr_data = gr_gp10b_init_ctxsw_hdr_data,
         },
         .fb = {
                 .reset = gv11b_fb_reset,