Merge tag 'for-5.17/dm-changes' of git://git.kernel.org/pub/scm/linux/kernel/git...

[platform/kernel/linux-starfive.git] / lib / objagg.c

diff --git a/lib/objagg.c b/lib/objagg.c
index 5e1676c..1e24862 100644 (file)
--- a/lib/objagg.c
+++ b/lib/objagg.c
@@ -781,7 +781,6 @@ static struct objagg_tmp_graph *objagg_tmp_graph_create(struct objagg *objagg)
        struct objagg_tmp_node *node;
        struct objagg_tmp_node *pnode;
        struct objagg_obj *objagg_obj;
-       size_t alloc_size;
        int i, j;
 
        graph = kzalloc(sizeof(*graph), GFP_KERNEL);
@@ -793,9 +792,7 @@ static struct objagg_tmp_graph *objagg_tmp_graph_create(struct objagg *objagg)
                goto err_nodes_alloc;
        graph->nodes_count = nodes_count;
 
-       alloc_size = BITS_TO_LONGS(nodes_count * nodes_count) *
-                    sizeof(unsigned long);
-       graph->edges = kzalloc(alloc_size, GFP_KERNEL);
+       graph->edges = bitmap_zalloc(nodes_count * nodes_count, GFP_KERNEL);
        if (!graph->edges)
                goto err_edges_alloc;
 
@@ -833,7 +830,7 @@ err_nodes_alloc:
 
 static void objagg_tmp_graph_destroy(struct objagg_tmp_graph *graph)
 {
-       kfree(graph->edges);
+       bitmap_free(graph->edges);
        kfree(graph->nodes);
        kfree(graph);
 }