1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index 2e57d01e..b642075e 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -92,7 +92,10 @@ int gk20a_init_mm_setup_hw(struct gk20a *g)
 
         nvgpu_log_fn(g, " ");
 
-        g->ops.fb.set_mmu_page_size(g);
+        if (g->ops.fb.set_mmu_page_size) {
+                g->ops.fb.set_mmu_page_size(g);
+        }
+
         if (g->ops.fb.set_use_full_comp_tag_line) {
                 mm->use_full_comp_tag_line =
                         g->ops.fb.set_use_full_comp_tag_line(g);