The number of chunks/chunks_array may be passed in
by userspace and can be large.

Signed-off-by: Chen Li <[email protected]>
---
 drivers/gpu/drm/radeon/radeon_cs.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/drivers/gpu/drm/radeon/radeon_cs.c 
b/drivers/gpu/drm/radeon/radeon_cs.c
index 35e937d39b51..fb736ef9f9aa 100644
--- a/drivers/gpu/drm/radeon/radeon_cs.c
+++ b/drivers/gpu/drm/radeon/radeon_cs.c
@@ -288,7 +288,7 @@ int radeon_cs_parser_init(struct radeon_cs_parser *p, void 
*data)
        p->chunk_relocs = NULL;
        p->chunk_flags = NULL;
        p->chunk_const_ib = NULL;
-       p->chunks_array = kcalloc(cs->num_chunks, sizeof(uint64_t), GFP_KERNEL);
+       p->chunks_array = kvmalloc_array(cs->num_chunks, sizeof(uint64_t), 
GFP_KERNEL);
        if (p->chunks_array == NULL) {
                return -ENOMEM;
        }
@@ -299,7 +299,7 @@ int radeon_cs_parser_init(struct radeon_cs_parser *p, void 
*data)
        }
        p->cs_flags = 0;
        p->nchunks = cs->num_chunks;
-       p->chunks = kcalloc(p->nchunks, sizeof(struct radeon_cs_chunk), 
GFP_KERNEL);
+       p->chunks = kvmalloc_array(p->nchunks, sizeof(struct radeon_cs_chunk), 
GFP_KERNEL);
        if (p->chunks == NULL) {
                return -ENOMEM;
        }
@@ -452,8 +452,8 @@ static void radeon_cs_parser_fini(struct radeon_cs_parser 
*parser, int error, bo
        kvfree(parser->vm_bos);
        for (i = 0; i < parser->nchunks; i++)
                kvfree(parser->chunks[i].kdata);
-       kfree(parser->chunks);
-       kfree(parser->chunks_array);
+       kvfree(parser->chunks);
+       kvfree(parser->chunks_array);
        radeon_ib_free(parser->rdev, &parser->ib);
        radeon_ib_free(parser->rdev, &parser->const_ib);
 }
-- 
2.30.0



_______________________________________________
amd-gfx mailing list
[email protected]
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to