2 files changed, 6 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/enabled.h b/drivers/gpu/nvgpu/include/nvgpu/enabled.h
index fc7bab90..77b54ab9 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/enabled.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/enabled.h
@@ -181,10 +181,13 @@ struct gk20a;
 /* PLATFORM_ATOMIC support */
 #define NVGPU_SUPPORT_PLATFORM_ATOMIC           71
 
+/* NVGPU_GPU_IOCTL_SET_MMU_DEBUG_MODE is available */
+#define NVGPU_SUPPORT_SET_CTX_MMU_DEBUG_MODE    72
+
 /*
  * Must be greater than the largest bit offset in the above list.
  */
-#define NVGPU_MAX_ENABLED_BITS                  72
+#define NVGPU_MAX_ENABLED_BITS                  73U
 
 /**
  * nvgpu_is_enabled - Check if the passed flag is enabled.
diff --git a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
index a0af0c5c..f3a83602 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
@@ -529,6 +529,8 @@ struct gpu_ops {
                         u32 num_ppcs, u32 reg_list_ppc_count,
                         u32 *__offset_in_segment);
                 void (*set_debug_mode)(struct gk20a *g, bool enable);
+                int (*set_mmu_debug_mode)(struct gk20a *g,
+                        struct channel_gk20a *ch, bool enable);
         } gr;
         struct {
                 void (*init_hw)(struct gk20a *g);