diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
index 5838a5785..cc880abe2 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
@@ -35,10 +35,6 @@
 
 #define GK20A_TIMEOUT_FPGA		100000U /* 100 sec */
 
-/* Flags to be passed to g->ops.gr.alloc_obj_ctx() */
-#define NVGPU_OBJ_CTX_FLAGS_SUPPORT_GFXP		BIT32(1)
-#define NVGPU_OBJ_CTX_FLAGS_SUPPORT_CILP		BIT32(2)
-
 struct tsg_gk20a;
 struct nvgpu_gr_ctx;
 struct channel_gk20a;
diff --git a/drivers/gpu/nvgpu/include/nvgpu/gr/obj_ctx.h b/drivers/gpu/nvgpu/include/nvgpu/gr/obj_ctx.h
index 42e1cc7a5..9f620be84 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gr/obj_ctx.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gr/obj_ctx.h
@@ -26,6 +26,9 @@
 #include <nvgpu/types.h>
 #include <nvgpu/lock.h>
 
+#define NVGPU_OBJ_CTX_FLAGS_SUPPORT_GFXP		BIT32(1)
+#define NVGPU_OBJ_CTX_FLAGS_SUPPORT_CILP		BIT32(2)
+
 struct gk20a;
 struct nvgpu_gr_ctx;
 struct nvgpu_gr_subctx;
diff --git a/drivers/gpu/nvgpu/os/linux/ioctl_channel.c b/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
index f79b0be57..01b00aa70 100644
--- a/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
+++ b/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
@@ -40,6 +40,7 @@
 #include <nvgpu/channel_sync_syncpt.h>
 #include <nvgpu/runlist.h>
 #include <nvgpu/gr/ctx.h>
+#include <nvgpu/gr/obj_ctx.h>
 #include <nvgpu/fence.h>
 
 #include "platform_gk20a.h"