diff --git a/drivers/gpu/nvgpu/Makefile b/drivers/gpu/nvgpu/Makefile
index 8d8fed711..0ed7d9109 100644
--- a/drivers/gpu/nvgpu/Makefile
+++ b/drivers/gpu/nvgpu/Makefile
@@ -25,6 +25,8 @@ ifeq ($(CONFIG_NVGPU_LS_PMU),y)
 ccflags-y += -DNVGPU_LS_PMU
 endif
 
+ccflags-y += -DNVGPU_ENGINE
+
 obj-$(CONFIG_GK20A) := nvgpu.o
 
 # OS independent parts of nvgpu. The work to collect files here
diff --git a/drivers/gpu/nvgpu/Makefile.tmk b/drivers/gpu/nvgpu/Makefile.tmk
index 130decbf2..e2dd9fe3f 100644
--- a/drivers/gpu/nvgpu/Makefile.tmk
+++ b/drivers/gpu/nvgpu/Makefile.tmk
@@ -67,6 +67,7 @@ NVGPU_DEBUGGER                  := 1
 NVGPU_LS_PMU                    := 1
 NV_COMPONENT_CFLAGS             += -DNVGPU_DEBUGGER
 NV_COMPONENT_CFLAGS             += -DNVGPU_LS_PMU
+NV_COMPONENT_CFLAGS             += -DNVGPU_ENGINE
 endif
 
 _NV_TOOLCHAIN_CFLAGS            += -rdynamic -g