qdev = (struct qxl_device *)bo->gem_base.dev->dev_private;
 
        qxl_surface_evict(qdev, bo, false);
+       WARN_ON_ONCE(bo->map_count > 0);
        mutex_lock(&qdev->gem.mutex);
        list_del_init(&bo->list);
        mutex_unlock(&qdev->gem.mutex);
        if (bo->kptr) {
                if (ptr)
                        *ptr = bo->kptr;
+               bo->map_count++;
                return 0;
        }
        r = ttm_bo_kmap(&bo->tbo, 0, bo->tbo.num_pages, &bo->kmap);
        bo->kptr = ttm_kmap_obj_virtual(&bo->kmap, &is_iomem);
        if (ptr)
                *ptr = bo->kptr;
+       bo->map_count = 1;
        return 0;
 }
 
 {
        if (bo->kptr == NULL)
                return;
+       bo->map_count--;
+       if (bo->map_count > 0)
+               return;
        bo->kptr = NULL;
        ttm_bo_kunmap(&bo->kmap);
 }
 
  * Authors: Andreas Pokorny
  */
 
-#include "qxl_drv.h"
+#include "qxl_object.h"
 
 /* Empty Implementations as there should not be any other driver for a virtual
  * device that might share buffers with qxl */
 
 void *qxl_gem_prime_vmap(struct drm_gem_object *obj)
 {
-       WARN_ONCE(1, "not implemented");
-       return ERR_PTR(-ENOSYS);
+       struct qxl_bo *bo = gem_to_qxl_bo(obj);
+       void *ptr;
+       int ret;
+
+       ret = qxl_bo_kmap(bo, &ptr);
+       if (ret < 0)
+               return ERR_PTR(ret);
+
+       return ptr;
 }
 
 void qxl_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
 {
-       WARN_ONCE(1, "not implemented");
+       struct qxl_bo *bo = gem_to_qxl_bo(obj);
+
+       qxl_bo_kunmap(bo);
 }
 
 int qxl_gem_prime_mmap(struct drm_gem_object *obj,