ia_css_ptr hmm_alloc(size_t bytes);
 ia_css_ptr hmm_create_from_userdata(size_t bytes, const void __user *userptr);
+ia_css_ptr hmm_create_from_vmalloc_buf(size_t bytes, void *vmalloc_addr);
+
 void hmm_free(ia_css_ptr ptr);
 int hmm_load(ia_css_ptr virt, void *data, unsigned int bytes);
 int hmm_store(ia_css_ptr virt, const void *data, unsigned int bytes);
 
 
 enum hmm_bo_type {
        HMM_BO_PRIVATE,
+       HMM_BO_VMALLOC,
        HMM_BO_USER,
        HMM_BO_LAST,
 };
  */
 int hmm_bo_alloc_pages(struct hmm_buffer_object *bo,
                       enum hmm_bo_type type,
-                      const void __user *userptr);
+                      const void __user *userptr,
+                      void *vmalloc_addr);
 void hmm_bo_free_pages(struct hmm_buffer_object *bo);
 int hmm_bo_page_allocated(struct hmm_buffer_object *bo);
 
 
 
 /*
  * p: private
- * s: shared
+ * v: vmalloc
  * u: user
- * i: ion
  */
-static const char hmm_bo_type_string[] = "psui";
+static const char hmm_bo_type_string[] = "pvu";
 
 static ssize_t bo_show(struct device *dev, struct device_attribute *attr,
                       char *buf, struct list_head *bo_list, bool active)
        hmm_initialized = false;
 }
 
-static ia_css_ptr __hmm_alloc(size_t bytes, enum hmm_bo_type type, const void __user *userptr)
+static ia_css_ptr __hmm_alloc(size_t bytes, enum hmm_bo_type type,
+                             const void __user *userptr,
+                             void *vmalloc_addr)
 {
        unsigned int pgnr;
        struct hmm_buffer_object *bo;
        }
 
        /* Allocate pages for memory */
-       ret = hmm_bo_alloc_pages(bo, type, userptr);
+       ret = hmm_bo_alloc_pages(bo, type, userptr, vmalloc_addr);
        if (ret) {
                dev_err(atomisp_dev, "hmm_bo_alloc_pages failed.\n");
                goto alloc_page_err;
 
 ia_css_ptr hmm_alloc(size_t bytes)
 {
-       return __hmm_alloc(bytes, HMM_BO_PRIVATE, NULL);
+       return __hmm_alloc(bytes, HMM_BO_PRIVATE, NULL, NULL);
+}
+
+ia_css_ptr hmm_create_from_vmalloc_buf(size_t bytes, void *vmalloc_addr)
+{
+       return __hmm_alloc(bytes, HMM_BO_VMALLOC, NULL, vmalloc_addr);
 }
 
 ia_css_ptr hmm_create_from_userdata(size_t bytes, const void __user *userptr)
 {
-       return __hmm_alloc(bytes, HMM_BO_USER, userptr);
+       return __hmm_alloc(bytes, HMM_BO_USER, userptr, NULL);
 }
 
 void hmm_free(ia_css_ptr virt)
 
        return -ENOMEM;
 }
 
+static int alloc_vmalloc_pages(struct hmm_buffer_object *bo, void *vmalloc_addr)
+{
+       void *vaddr = vmalloc_addr;
+       int i;
+
+       for (i = 0; i < bo->pgnr; i++) {
+               bo->pages[i] = vmalloc_to_page(vaddr);
+               if (!bo->pages[i]) {
+                       dev_err(atomisp_dev, "Error could not get page %d of vmalloc buf\n", i);
+                       return -ENOMEM;
+               }
+               vaddr += PAGE_SIZE;
+       }
+
+       return 0;
+}
+
 /*
  * allocate/free physical pages for the bo.
  *
  * type indicate where are the pages from. currently we have 3 types
- * of memory: HMM_BO_PRIVATE, HMM_BO_USER.
+ * of memory: HMM_BO_PRIVATE, HMM_BO_VMALLOC, HMM_BO_USER.
+ *
+ * vmalloc_addr is only valid when type is HMM_BO_VMALLOC.
  *
  * userptr is only valid when type is HMM_BO_USER, it indicates
  * the start address from user space task.
  */
 int hmm_bo_alloc_pages(struct hmm_buffer_object *bo,
                       enum hmm_bo_type type,
-                      const void __user *userptr)
+                      const void __user *userptr,
+                      void *vmalloc_addr)
 {
        int ret = -EINVAL;
 
                goto alloc_err;
        }
 
-       /*
-        * TO DO:
-        * add HMM_BO_USER type
-        */
        if (type == HMM_BO_PRIVATE) {
                ret = alloc_private_pages(bo);
+       } else if (type == HMM_BO_VMALLOC) {
+               ret = alloc_vmalloc_pages(bo, vmalloc_addr);
        } else if (type == HMM_BO_USER) {
                ret = alloc_user_pages(bo, userptr);
        } else {
 
        if (bo->type == HMM_BO_PRIVATE)
                free_private_bo_pages(bo);
+       else if (bo->type == HMM_BO_VMALLOC)
+               ; /* No-op, nothing to do */
        else if (bo->type == HMM_BO_USER)
                free_user_pages(bo, bo->pgnr);
        else