diff --git a/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c b/drivers/gpu/nvgpu/common/mm/nvgpu_allocator.c
index a3c4e3b60..211b353bb 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 4eb769852..3579b0fb5 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 {						\