arm64: mm: Use better bitmap_zalloc()

Use better bitmap_zalloc() to allocate bitmap.

Cc: Catalin Marinas <catalin.marinas@arm.com>
Cc: Will Deacon <will@kernel.org>
Cc: linux-arm-kernel@lists.infradead.org
Signed-off-by: Kefeng Wang <wangkefeng.wang@huawei.com>
Link: https://lore.kernel.org/r/20210529111510.186355-1-wangkefeng.wang@huawei.com
Signed-off-by: Will Deacon <will@kernel.org>

diff --git a/arch/arm64/mm/context.c b/arch/arm64/mm/context.c
index 001737a..cd72576 100644 (file)
--- a/arch/arm64/mm/context.c
+++ b/arch/arm64/mm/context.c
@@ -402,14 +402,12 @@ static int asids_init(void)
 {
        asid_bits = get_cpu_asid_bits();
        atomic64_set(&asid_generation, ASID_FIRST_VERSION);
-       asid_map = kcalloc(BITS_TO_LONGS(NUM_USER_ASIDS), sizeof(*asid_map),
-                          GFP_KERNEL);
+       asid_map = bitmap_zalloc(NUM_USER_ASIDS, GFP_KERNEL);
        if (!asid_map)
                panic("Failed to allocate bitmap for %lu ASIDs\n",
                      NUM_USER_ASIDS);
 
-       pinned_asid_map = kcalloc(BITS_TO_LONGS(NUM_USER_ASIDS),
-                                 sizeof(*pinned_asid_map), GFP_KERNEL);
+       pinned_asid_map = bitmap_zalloc(NUM_USER_ASIDS, GFP_KERNEL);
        nr_pinned_asids = 0;
 
        /*