1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
index c5390fe4..a67b1476 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
@@ -1657,8 +1657,10 @@ static void gv11b_write_bundle_veid_state(struct gk20a *g, u32 index)
         struct av_list_gk20a *sw_veid_bundle_init =
                         &g->gr.ctx_vars.sw_veid_bundle_init;
         u32 j;
-        u32 num_subctx = nvgpu_get_litter_value(g, GPU_LIT_NUM_SUBCTX);
-        u32 err = 0;
+        u32 data = gk20a_readl(g, gr_pri_fe_chip_def_info_r());
+        u32  num_subctx, err = 0;
+
+        num_subctx = gr_pri_fe_chip_def_info_max_veid_count_v(data);
 
         for (j = 0; j < num_subctx; j++) {