/* Copyright (c) 2019 HiSilicon Limited. */
 #include <crypto/internal/acompress.h>
 #include <linux/bitfield.h>
+#include <linux/bitmap.h>
 #include <linux/dma-mapping.h>
 #include <linux/scatterlist.h>
 #include "zip.h"
                req_q = &ctx->qp_ctx[i].req_q;
                req_q->size = QM_Q_DEPTH;
 
-               req_q->req_bitmap = kcalloc(BITS_TO_LONGS(req_q->size),
-                                           sizeof(long), GFP_KERNEL);
+               req_q->req_bitmap = bitmap_zalloc(req_q->size, GFP_KERNEL);
                if (!req_q->req_bitmap) {
                        ret = -ENOMEM;
                        if (i == 0)
        return 0;
 
 err_free_loop1:
-       kfree(ctx->qp_ctx[HZIP_QPC_DECOMP].req_q.req_bitmap);
+       bitmap_free(ctx->qp_ctx[HZIP_QPC_DECOMP].req_q.req_bitmap);
 err_free_loop0:
        kfree(ctx->qp_ctx[HZIP_QPC_COMP].req_q.q);
 err_free_bitmap:
-       kfree(ctx->qp_ctx[HZIP_QPC_COMP].req_q.req_bitmap);
+       bitmap_free(ctx->qp_ctx[HZIP_QPC_COMP].req_q.req_bitmap);
        return ret;
 }
 
 
        for (i = 0; i < HZIP_CTX_Q_NUM; i++) {
                kfree(ctx->qp_ctx[i].req_q.q);
-               kfree(ctx->qp_ctx[i].req_q.req_bitmap);
+               bitmap_free(ctx->qp_ctx[i].req_q.req_bitmap);
        }
 }