1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 4f897b97..d24b8bf7 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -3473,6 +3473,8 @@ static int gr_gk20a_init_gr_config(struct gk20a *g, struct gr_gk20a *gr)
                 gr->sm_to_cluster = nvgpu_kzalloc(g, gr->gpc_count *
                                         gr->max_tpc_per_gpc_count *
                                         sm_per_tpc * sizeof(struct sm_info));
+                if (!gr->sm_to_cluster)
+                        goto clean_up;
         } else {
                 memset(gr->sm_to_cluster, 0, gr->gpc_count *
                                         gr->max_tpc_per_gpc_count *