1 files changed, 4 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/debug_ce.c b/drivers/gpu/nvgpu/common/linux/debug_ce.c
index 1c2d4170..cea0bb47 100644
--- a/drivers/gpu/nvgpu/common/linux/debug_ce.c
+++ b/drivers/gpu/nvgpu/common/linux/debug_ce.c
@@ -13,19 +13,18 @@
  */
 
 #include "debug_ce.h"
-#include "gk20a/platform_gk20a.h"
 #include "os_linux.h"
 
 #include <linux/debugfs.h>
 
 void gk20a_ce_debugfs_init(struct gk20a *g)
 {
-        struct gk20a_platform *platform = dev_get_drvdata(dev_from_gk20a(g));
+        struct nvgpu_os_linux *l = nvgpu_os_linux_from_gk20a(g);
 
         debugfs_create_u32("ce_app_ctx_count", S_IWUSR | S_IRUGO,
-                           platform->debugfs, &g->ce_app.ctx_count);
+                           l->debugfs, &g->ce_app.ctx_count);
         debugfs_create_u32("ce_app_state", S_IWUSR | S_IRUGO,
-                           platform->debugfs, &g->ce_app.app_state);
+                           l->debugfs, &g->ce_app.app_state);
         debugfs_create_u32("ce_app_next_ctx_id", S_IWUSR | S_IRUGO,
-                           platform->debugfs, &g->ce_app.next_ctx_id);
+                           l->debugfs, &g->ce_app.next_ctx_id);
 }