diff --git a/drivers/video/tegra/nvmap/nvmap_alloc.c b/drivers/video/tegra/nvmap/nvmap_alloc.c
index 7f3a64a4..6b219d08 100644
--- a/drivers/video/tegra/nvmap/nvmap_alloc.c
+++ b/drivers/video/tegra/nvmap/nvmap_alloc.c
@@ -16,6 +16,7 @@
 #include <linux/libnvdimm.h>
 #include <linux/rtmutex.h>
 #include <linux/slab.h>
+#include <linux/vmalloc.h>
 #include "nvmap_stats.h"
 #include "nvmap_dev.h"
 #include "nvmap_alloc.h"
@@ -581,4 +582,4 @@ void nvmap_alloc_free(struct page **pages, unsigned int nr_page, bool from_va,
 phys_addr_t nvmap_alloc_get_co_base(struct nvmap_handle *h)
 {
 	return h->carveout->base;
-}
\ No newline at end of file
+}
diff --git a/drivers/video/tegra/nvmap/nvmap_handle.c b/drivers/video/tegra/nvmap/nvmap_handle.c
index b5f191d0..74b0fbd3 100644
--- a/drivers/video/tegra/nvmap/nvmap_handle.c
+++ b/drivers/video/tegra/nvmap/nvmap_handle.c
@@ -20,6 +20,7 @@
 #include <linux/slab.h>
 #include <linux/nvmap.h>
 #include <linux/version.h>
+#include <linux/vmalloc.h>
 #include <linux/wait.h>
 #include <soc/tegra/fuse.h>
 #include <asm/pgtable.h>
diff --git a/drivers/video/tegra/nvmap/nvmap_heap.c b/drivers/video/tegra/nvmap/nvmap_heap.c
index 77a127bc..f7d8c078 100644
--- a/drivers/video/tegra/nvmap/nvmap_heap.c
+++ b/drivers/video/tegra/nvmap/nvmap_heap.c
@@ -26,6 +26,7 @@
 #include <linux/dma-mapping.h>
 #include <linux/dma-map-ops.h>
 #include <linux/rtmutex.h>
+#include <linux/vmalloc.h>
 #include "nvmap_dev.h"
 #include "nvmap_alloc.h"
 #include "nvmap_handle.h"