1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
index 4c2b00a8..3d99e94d 100644
--- a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
@@ -533,10 +533,15 @@ static int gr_gm20b_ctx_state_floorsweep(struct gk20a *g)
                                         + gpc_offset + tpc_offset,
                                 gr_gpc0_tpc0_pe_cfg_smid_value_f(sm_id));
 
+                        g->gr.sm_to_cluster[sm_id].tpc_index = tpc_index;
+                        g->gr.sm_to_cluster[sm_id].gpc_index = gpc_index;
+
                         sm_id++;
                 }
         }
 
+        gr->no_of_sm = sm_id;
+
         for (gpc_index = 0; gpc_index < gr->gpc_count; gpc_index++)
                 tpc_per_gpc |= gr->gpc_tpc_count[gpc_index]
                              << (gr_pd_num_tpc_per_gpc__size_1_v() * gpc_index);