2 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.c b/drivers/gpu/nvgpu/vgpu/vgpu.c
index bd52dc0a..c8929e93 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.c
@@ -677,7 +677,7 @@ int vgpu_probe(struct platform_device *pdev)
                 return err;
         }
 
-        gk20a->dev = dev;
+        l->dev = dev;
         if (tegra_platform_is_linsim() || tegra_platform_is_vdk())
                 __nvgpu_set_enabled(gk20a, NVGPU_IS_FMODEL, true);
 
diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.h b/drivers/gpu/nvgpu/vgpu/vgpu.h
index 92b82ac4..cee81bb1 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.h
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.h
@@ -20,6 +20,7 @@
 #include <linux/tegra_vgpu.h>
 #include "gk20a/gk20a.h"
 #include "gk20a/platform_gk20a.h"
+#include "common/linux/os_linux.h"
 
 #include <nvgpu/thread.h>