1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
index 1336557a..4280e8ff 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
@@ -48,7 +48,6 @@
 #include "gv11b/subctx_gv11b.h"
 #include "gv11b/gv11b.h"
 #include "gv11b/gr_pri_gv11b.h"
-#include "gv11b/fb_gv11b.h"
 
 #include <nvgpu/hw/gv11b/hw_gr_gv11b.h>
 #include <nvgpu/hw/gv11b/hw_fifo_gv11b.h>
@@ -3741,8 +3740,9 @@ int gv11b_gr_wait_for_sm_lock_down(struct gk20a *g,
                         return 0;
                 }
 
-                if (mmu_debug_mode_enabled) {
-                        gv11b_fb_handle_replayable_mmu_fault(g);
+                if (mmu_debug_mode_enabled &&
+                    g->ops.fb.handle_replayable_fault != NULL) {
+                        g->ops.fb.handle_replayable_fault(g);
                 } else {
                         /* if an mmu fault is pending and mmu debug mode is not
                          * enabled, the sm will never lock down.