2 files changed, 5 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c b/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c
index a3c4e3b6..211b353b 100644
--- a/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c
+++ b/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c
@@ -201,9 +201,9 @@ void nvgpu_fini_alloc_debug(struct nvgpu_allocator *a)
 #endif
 }
 
+#ifdef CONFIG_DEBUG_FS
 void nvgpu_alloc_debugfs_init(struct device *dev)
 {
-#ifdef CONFIG_DEBUG_FS
         struct gk20a_platform *platform = dev_get_drvdata(dev);
         struct dentry *gpu_root = platform->debugfs;
         struct gk20a *g = get_gk20a(dev);
@@ -214,5 +214,5 @@ void nvgpu_alloc_debugfs_init(struct device *dev)
 
         debugfs_create_u32("tracing", 0664, g->debugfs_allocators,
                            &nvgpu_alloc_tracing_on);
-#endif
 }
+#endif
diff --git a/drivers/gpu/nvgpu/include/nvgpu/allocator.h b/drivers/gpu/nvgpu/include/nvgpu/allocator.h
index 4eb76985..3579b0fb 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/allocator.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/allocator.h
@@ -281,7 +281,10 @@ static inline void nvgpu_alloc_disable_dbg(struct nvgpu_allocator *a)
  */
 extern u32 nvgpu_alloc_tracing_on;
 
+#ifdef CONFIG_DEBUG_FS
+struct device;
 void nvgpu_alloc_debugfs_init(struct device *dev);
+#endif
 
 #define nvgpu_alloc_trace_func()                        \
         do {                                            \