static int vmlfb_mmap(struct fb_info *info, struct vm_area_struct *vma)
 {
        struct vml_info *vinfo = container_of(info, struct vml_info, info);
-       unsigned long size = vma->vm_end - vma->vm_start;
        unsigned long offset = vma->vm_pgoff << PAGE_SHIFT;
        int ret;
 
-       if (vma->vm_pgoff > (~0UL >> PAGE_SHIFT))
-               return -EINVAL;
-       if (offset + size > vinfo->vram_contig_size)
-               return -EINVAL;
        ret = vmlfb_vram_offset(vinfo, offset);
        if (ret)
                return -EINVAL;
-       offset += vinfo->vram_start;
+
        pgprot_val(vma->vm_page_prot) |= _PAGE_PCD;
        pgprot_val(vma->vm_page_prot) &= ~_PAGE_PWT;
-       if (remap_pfn_range(vma, vma->vm_start, offset >> PAGE_SHIFT,
-                                               size, vma->vm_page_prot))
-               return -EAGAIN;
-       return 0;
+
+       return vm_iomap_memory(vma, vinfo->vram_start,
+                       vinfo->vram_contig_size);
 }
 
 static int vmlfb_sync(struct fb_info *info)