1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp106/hal_gp106.c b/drivers/gpu/nvgpu/gp106/hal_gp106.c
index d1b60af5..43b1d2e0 100644
--- a/drivers/gpu/nvgpu/gp106/hal_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/hal_gp106.c
@@ -97,6 +97,7 @@
 #include <nvgpu/hw/gp106/hw_top_gp106.h>
 #include <nvgpu/hw/gp106/hw_pram_gp106.h>
 #include <nvgpu/hw/gp106/hw_pwr_gp106.h>
+#include <nvgpu/hw/gp106/hw_gr_gp106.h>
 
 
 static int gp106_get_litter_value(struct gk20a *g, int value)
@@ -387,6 +388,7 @@ static const struct gpu_ops gp106_ops = {
                 .get_pmm_per_chiplet_offset =
                         gr_gm20b_get_pmm_per_chiplet_offset,
                 .split_fbpa_broadcast_addr = gr_gk20a_split_fbpa_broadcast_addr,
+                .fecs_ctxsw_mailbox_size = gr_fecs_ctxsw_mailbox__size_1_v,
         },
         .fb = {
                 .reset = gp106_fb_reset,