1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
index f13c2735..5ae6b638 100644
--- a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
@@ -332,6 +332,10 @@ static const struct gpu_ops gp10b_ops = {
                 .create_gr_sysfs = gr_gp10b_create_sysfs,
                 .set_ctxsw_preemption_mode = gr_gp10b_set_ctxsw_preemption_mode,
                 .init_ctxsw_hdr_data = gr_gp10b_init_ctxsw_hdr_data,
+                .init_gfxp_wfi_timeout_count =
+                                gr_gp10b_init_gfxp_wfi_timeout_count,
+                .get_max_gfxp_wfi_timeout_count =
+                                gr_gp10b_get_max_gfxp_wfi_timeout_count,
         },
         .fb = {
                 .reset = fb_gk20a_reset,