1 files changed, 2 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/module.c b/drivers/gpu/nvgpu/common/linux/module.c
index cc2b5ec7..b95bac0b 100644
--- a/drivers/gpu/nvgpu/common/linux/module.c
+++ b/drivers/gpu/nvgpu/common/linux/module.c
@@ -52,9 +52,6 @@
 #include "sim.h"
 #ifdef CONFIG_TEGRA_19x_GPU
 #include "nvgpu_gpuid_t19x.h"
-#ifdef CONFIG_TEGRA_GR_VIRTUALIZATION
-#include "vgpu/vgpu_t19x.h"
-#endif
 #endif
 
 #include "os_linux.h"
@@ -389,8 +386,8 @@ static struct of_device_id tegra_gk20a_of_match[] = {
         { .compatible = TEGRA_19x_GPU_COMPAT_TEGRA,
                 .data = &t19x_gpu_tegra_platform },
 #ifdef CONFIG_TEGRA_GR_VIRTUALIZATION
-        { .compatible = TEGRA_19x_VGPU_COMPAT_TEGRA,
-                .data = &t19x_vgpu_tegra_platform },
+        { .compatible = "nvidia,gv11b-vgpu",
+                .data = &gv11b_vgpu_tegra_platform},
 #endif
 #endif
 #ifdef CONFIG_TEGRA_GR_VIRTUALIZATION