1 files changed, 3 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/debug_gr.c b/drivers/gpu/nvgpu/common/linux/debug_gr.c
index 732c2a7e..d54c6d63 100644
--- a/drivers/gpu/nvgpu/common/linux/debug_gr.c
+++ b/drivers/gpu/nvgpu/common/linux/debug_gr.c
@@ -13,18 +13,17 @@
  */
 
 #include "debug_gr.h"
-#include "gk20a/platform_gk20a.h"
 #include "os_linux.h"
 
 #include <linux/debugfs.h>
 
 int gr_gk20a_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);
 
-        g->debugfs_gr_default_attrib_cb_size =
+        l->debugfs_gr_default_attrib_cb_size =
                 debugfs_create_u32("gr_default_attrib_cb_size",
-                                   S_IRUGO|S_IWUSR, platform->debugfs,
+                                   S_IRUGO|S_IWUSR, l->debugfs,
                                    &g->gr.attrib_cb_default_size);
 
         return 0;