1 files changed, 25 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/fb/fb_gv100.c b/drivers/gpu/nvgpu/common/fb/fb_gv100.c
index 193cf2f0..508259e7 100644
--- a/drivers/gpu/nvgpu/common/fb/fb_gv100.c
+++ b/drivers/gpu/nvgpu/common/fb/fb_gv100.c
@@ -1,7 +1,7 @@
 /*
  * GV100 FB
  *
- * Copyright (c) 2017-2018, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2017-2020, NVIDIA CORPORATION.  All rights reserved.
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -300,3 +300,27 @@ size_t gv100_fb_get_vidmem_size(struct gk20a *g)
 
         return bytes;
 }
+
+void gv100_fb_set_mmu_debug_mode(struct gk20a *g, bool enable)
+{
+        u32 data, fb_ctrl, hsmmu_ctrl;
+
+        if (enable) {
+                fb_ctrl = fb_mmu_debug_ctrl_debug_enabled_f();
+                hsmmu_ctrl = fb_hsmmu_pri_mmu_debug_ctrl_debug_enabled_f();
+                g->mmu_debug_ctrl = true;
+        } else {
+                fb_ctrl = fb_mmu_debug_ctrl_debug_disabled_f();
+                hsmmu_ctrl = fb_hsmmu_pri_mmu_debug_ctrl_debug_disabled_f();
+                g->mmu_debug_ctrl = false;
+        }
+
+        data = nvgpu_readl(g, fb_mmu_debug_ctrl_r());
+        data = set_field(data, fb_mmu_debug_ctrl_debug_m(), fb_ctrl);
+        nvgpu_writel(g, fb_mmu_debug_ctrl_r(), data);
+
+        data = nvgpu_readl(g, fb_hsmmu_pri_mmu_debug_ctrl_r());
+        data = set_field(data,
+                        fb_hsmmu_pri_mmu_debug_ctrl_debug_m(), hsmmu_ctrl);
+        nvgpu_writel(g, fb_hsmmu_pri_mmu_debug_ctrl_r(), data);
+}