2 files changed, 4 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
index 4b1cf212..9fafa52f 100644
--- a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
@@ -30,11 +30,11 @@
 #include <nvgpu/vgpu/vgpu_ivc.h>
 #include <nvgpu/vgpu/vgpu.h>
 #include <nvgpu/channel.h>
+#include <nvgpu/tsg.h>
 
 #include "gr_vgpu.h"
 #include "gk20a/gk20a.h"
 #include "gk20a/dbg_gpu_gk20a.h"
-#include "gk20a/tsg_gk20a.h"
 #include "gk20a/fecs_trace_gk20a.h"
 
 #include <nvgpu/hw/gk20a/hw_gr_gk20a.h>
diff --git a/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c b/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
index a81b5022..3553bf51 100644
--- a/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
@@ -21,11 +21,12 @@
  */
 
 #include "gk20a/gk20a.h"
-#include "gk20a/tsg_gk20a.h"
 #include "fifo_vgpu.h"
-#include <nvgpu/channel.h>
 
+#include <nvgpu/channel.h>
+#include <nvgpu/tsg.h>
 #include <nvgpu/bug.h>
+
 #include <nvgpu/vgpu/tegra_vgpu.h>
 #include <nvgpu/vgpu/vgpu.h>