1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index 2fd393bd..24b22a7d 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -1071,9 +1071,9 @@ int gr_gp10b_alloc_gr_ctx(struct gk20a *g,
 
         (*gr_ctx)->t18x.ctx_id_valid = false;
 
-        if (flags & NVGPU_ALLOC_OBJ_FLAGS_GFXP)
+        if (flags & NVGPU_OBJ_CTX_FLAGS_SUPPORT_GFXP)
                 graphics_preempt_mode = NVGPU_GRAPHICS_PREEMPTION_MODE_GFXP;
-        if (flags & NVGPU_ALLOC_OBJ_FLAGS_CILP)
+        if (flags & NVGPU_OBJ_CTX_FLAGS_SUPPORT_CILP)
                 compute_preempt_mode = NVGPU_COMPUTE_PREEMPTION_MODE_CILP;
 
         if (graphics_preempt_mode || compute_preempt_mode) {