drm/scheduler: trivial error handling fix

Return -ENOMEM when allocating the rq_list fails.

Signed-off-by: Christian König <christian.koenig@amd.com>
Reviewed-by: Huang Rui <ray.huang@amd.com>
Reviewed-by: Andrey Grodzovsky <andrey.grodzovsky@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

diff --git a/drivers/gpu/drm/scheduler/gpu_scheduler.c b/drivers/gpu/drm/scheduler/gpu_scheduler.c
index f566405f49e3c3af9fad4fe3403b44d245a97add..85c1f95752cc47e6d46e3d8deb3c3c256c7f1357 100644 (file)
--- a/drivers/gpu/drm/scheduler/gpu_scheduler.c
+++ b/drivers/gpu/drm/scheduler/gpu_scheduler.c
@@ -191,6 +191,9 @@ int drm_sched_entity_init(struct drm_sched_entity *entity,
        entity->num_rq_list = num_rq_list;
        entity->rq_list = kcalloc(num_rq_list, sizeof(struct drm_sched_rq *),
                                GFP_KERNEL);
+       if (!entity->rq_list)
+               return -ENOMEM;
+
        for (i = 0; i < num_rq_list; ++i)
                entity->rq_list[i] = rq_list[i];
        entity->last_scheduled = NULL;