2 files changed, 6 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/channel.h b/drivers/gpu/nvgpu/include/nvgpu/channel.h
index e98af09b..764d047c 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/channel.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/channel.h
@@ -328,7 +328,11 @@ struct channel_gk20a {
         bool has_os_fence_framework_support;
 
         bool is_privileged_channel;
-        bool mmu_debug_mode_enabled;
+
+        /**
+         * MMU Debugger Mode is enabled for this channel if refcnt > 0
+         */
+        u32 mmu_debug_mode_refcnt;
 };
 
 static inline struct channel_gk20a *
diff --git a/drivers/gpu/nvgpu/include/nvgpu/tsg.h b/drivers/gpu/nvgpu/include/nvgpu/tsg.h
index b43a6b98..7cd97c97 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/tsg.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/tsg.h
@@ -127,7 +127,6 @@ gk20a_event_id_data_from_event_id_node(struct nvgpu_list_node *node)
                 ((uintptr_t)node - offsetof(struct gk20a_event_id_data, event_id_node));
 };
 
-int nvgpu_tsg_set_mmu_debug_mode(struct tsg_gk20a *tsg,
-                struct channel_gk20a *ch, bool enable);
+int nvgpu_tsg_set_mmu_debug_mode(struct channel_gk20a *ch, bool enable);
 
 #endif /* TSG_GK20A_H */