1 files changed, 3 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/hal_gm20b.c b/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
index 3b80ede0..51ddb561 100644
--- a/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
@@ -27,7 +27,6 @@
 #include "common/bus/bus_gk20a.h"
 #include "common/priv_ring/priv_ring_gm20b.h"
 #include "common/ptimer/ptimer_gk20a.h"
-#include "common/fb/fb_gk20a.h"
 #include "common/fb/fb_gm20b.h"
 #include "common/therm/therm_gm20b.h"
 #include "common/therm/therm_gm20b.h"
@@ -332,8 +331,8 @@ static const struct gpu_ops gm20b_ops = {
                 .set_debug_mode = gm20b_gr_set_debug_mode,
         },
         .fb = {
-                .reset = fb_gk20a_reset,
-                .init_hw = gk20a_fb_init_hw,
+                .reset = gm20b_fb_reset,
+                .init_hw = gm20b_fb_init_hw,
                 .init_fs_state = fb_gm20b_init_fs_state,
                 .set_mmu_page_size = gm20b_fb_set_mmu_page_size,
                 .set_use_full_comp_tag_line =
@@ -350,7 +349,7 @@ static const struct gpu_ops gm20b_ops = {
                 .read_wpr_info = gm20b_fb_read_wpr_info,
                 .is_debug_mode_enabled = gm20b_fb_debug_mode_enabled,
                 .set_debug_mode = gm20b_fb_set_debug_mode,
-                .tlb_invalidate = gk20a_fb_tlb_invalidate,
+                .tlb_invalidate = gm20b_fb_tlb_invalidate,
                 .mem_unlock = NULL,
         },
         .clock_gating = {