1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index e3d4b973..271a6d0c 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -571,11 +571,12 @@ int gr_gp10b_alloc_gr_ctx(struct gk20a *g,
                 (*gr_ctx)->preempt_mode = flags;
         }
 
-        if (class == PASCAL_COMPUTE_A)
+        if (class == PASCAL_COMPUTE_A) {
                 if (flags == NVGPU_GR_PREEMPTION_MODE_CILP)
                         (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CILP;
                 else
                         (*gr_ctx)->preempt_mode = NVGPU_GR_PREEMPTION_MODE_CTA;
+        }
 
         gk20a_dbg_fn("done");