diff --git a/drivers/gpu/nvgpu/common/linux/vgpu/dbg_vgpu.c b/drivers/gpu/nvgpu/common/linux/vgpu/dbg_vgpu.c
index 90679cc73..11ebe84b6 100644
--- a/drivers/gpu/nvgpu/common/linux/vgpu/dbg_vgpu.c
+++ b/drivers/gpu/nvgpu/common/linux/vgpu/dbg_vgpu.c
@@ -17,8 +17,6 @@
 #include <nvgpu/vgpu/vgpu_ivc.h>
 #include <nvgpu/vgpu/tegra_vgpu.h>
 
-#include <uapi/linux/nvgpu.h>
-
 #include "gk20a/gk20a.h"
 #include "gk20a/channel_gk20a.h"
 #include "gk20a/dbg_gpu_gk20a.h"
@@ -87,12 +85,12 @@ int vgpu_dbg_set_powergate(struct dbg_session_gk20a *dbg_s, bool disable_powerga
 		if (dbg_s->is_pg_disabled)
 			return 0;
 		dbg_s->is_pg_disabled = true;
-		mode = NVGPU_DBG_GPU_POWERGATE_MODE_DISABLE;
+		mode = TEGRA_VGPU_POWERGATE_MODE_DISABLE;
 	} else {
 		if (!dbg_s->is_pg_disabled)
 			return 0;
 		dbg_s->is_pg_disabled = false;
-		mode = NVGPU_DBG_GPU_POWERGATE_MODE_ENABLE;
+		mode = TEGRA_VGPU_POWERGATE_MODE_ENABLE;
 	}
 
 	msg.cmd = TEGRA_VGPU_CMD_SET_POWERGATE;
diff --git a/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h b/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
index 882dd3421..a7a877ff4 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
@@ -425,6 +425,8 @@ struct tegra_vgpu_get_timestamps_zipper_params {
 	u32 source_id;
 };
 
+#define TEGRA_VGPU_POWERGATE_MODE_ENABLE	1
+#define TEGRA_VGPU_POWERGATE_MODE_DISABLE	2
 struct tegra_vgpu_set_powergate_params {
 	u32 mode;
 };