1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
index fee3e4ea..0037257c 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
@@ -17,6 +17,7 @@
  */
 
 #include "gk20a_allocator.h"
+#include <linux/vmalloc.h>
 
 /* init allocator struct */
 int gk20a_allocator_init(struct gk20a_allocator *allocator,
@@ -29,8 +30,7 @@ int gk20a_allocator_init(struct gk20a_allocator *allocator,
         allocator->base = start;
         allocator->limit = start + len - 1;
 
-        allocator->bitmap = kzalloc(BITS_TO_LONGS(len) * sizeof(long),
-                        GFP_KERNEL);
+        allocator->bitmap = vzalloc(BITS_TO_LONGS(len) * sizeof(long));
         if (!allocator->bitmap)
                 return -ENOMEM;
 
@@ -50,7 +50,7 @@ void gk20a_allocator_destroy(struct gk20a_allocator *allocator)
 {
         down_write(&allocator->rw_sema);
 
-        kfree(allocator->bitmap);
+        vfree(allocator->bitmap);
 
         memset(allocator, 0, sizeof(struct gk20a_allocator));
 }