struct ttm_resource *old_mem = &bo->mem;
        int r;
 
+       amdgpu_bo_move_notify(bo, evict, new_mem);
+
        /* Can't move a pinned BO */
        abo = ttm_to_amdgpu_bo(bo);
        if (WARN_ON_ONCE(abo->tbo.pin_count > 0))
            new_mem->mem_type == TTM_PL_SYSTEM) {
                r = ttm_bo_wait_ctx(bo, ctx);
                if (r)
-                       return r;
+                       goto fail;
 
                amdgpu_ttm_backend_unbind(bo->bdev, bo->ttm);
                ttm_resource_free(bo, &bo->mem);
                if (!amdgpu_mem_visible(adev, old_mem) ||
                    !amdgpu_mem_visible(adev, new_mem)) {
                        pr_err("Move buffer fallback to memcpy unavailable\n");
-                       return r;
+                       goto fail;
                }
 
                r = ttm_bo_move_memcpy(bo, ctx, new_mem);
                if (r)
-                       return r;
+                       goto fail;
        }
 
        if (bo->type == ttm_bo_type_device &&
        /* update statistics */
        atomic64_add((u64)bo->num_pages << PAGE_SHIFT, &adev->num_bytes_moved);
        return 0;
+fail:
+       swap(*new_mem, bo->mem);
+       amdgpu_bo_move_notify(bo, false, new_mem);
+       swap(*new_mem, bo->mem);
+       return r;
 }
 
 /**
 
                                       struct ttm_operation_ctx *ctx,
                                       struct ttm_resource *new_mem)
 {
-       return ttm_bo_move_memcpy(&gbo->bo, ctx, new_mem);
+       int ret;
+
+       drm_gem_vram_bo_driver_move_notify(gbo, evict, new_mem);
+       ret = ttm_bo_move_memcpy(&gbo->bo, ctx, new_mem);
+       if (ret) {
+               swap(*new_mem, gbo->bo.mem);
+               drm_gem_vram_bo_driver_move_notify(gbo, false, new_mem);
+               swap(*new_mem, gbo->bo.mem);
+       }
+       return ret;
 }
 
 /*
 
        struct nouveau_drm_tile *new_tile = NULL;
        int ret = 0;
 
+       nouveau_bo_move_ntfy(bo, evict, new_reg);
        ret = ttm_bo_wait_ctx(bo, ctx);
        if (ret)
-               return ret;
+               goto out_ntfy;
 
        if (nvbo->bo.pin_count)
                NV_WARN(drm, "Moving pinned object %p!\n", nvbo);
        if (drm->client.device.info.family < NV_DEVICE_INFO_V0_TESLA) {
                ret = nouveau_bo_vm_bind(bo, new_reg, &new_tile);
                if (ret)
-                       return ret;
+                       goto out_ntfy;
        }
 
        /* Fake bo copy. */
                else
                        nouveau_bo_vm_cleanup(bo, new_tile, &nvbo->tile);
        }
-
+out_ntfy:
+       if (ret) {
+               swap(*new_reg, bo->mem);
+               nouveau_bo_move_ntfy(bo, false, new_reg);
+               swap(*new_reg, bo->mem);
+       }
        return ret;
 }
 
 
        return ttm;
 }
 
-static int qxl_bo_move(struct ttm_buffer_object *bo, bool evict,
-                      struct ttm_operation_ctx *ctx,
-                      struct ttm_resource *new_mem)
-{
-       struct ttm_resource *old_mem = &bo->mem;
-       int ret;
-
-       ret = ttm_bo_wait_ctx(bo, ctx);
-       if (ret)
-               return ret;
-
-       if (old_mem->mem_type == TTM_PL_SYSTEM && bo->ttm == NULL) {
-               ttm_bo_move_null(bo, new_mem);
-               return 0;
-       }
-       return ttm_bo_move_memcpy(bo, ctx, new_mem);
-}
-
 static void qxl_bo_move_notify(struct ttm_buffer_object *bo,
                               bool evict,
                               struct ttm_resource *new_mem)
                qxl_surface_evict(qdev, qbo, new_mem ? true : false);
 }
 
+static int qxl_bo_move(struct ttm_buffer_object *bo, bool evict,
+                      struct ttm_operation_ctx *ctx,
+                      struct ttm_resource *new_mem)
+{
+       struct ttm_resource *old_mem = &bo->mem;
+       int ret;
+
+       qxl_bo_move_notify(bo, evict, new_mem);
+
+       ret = ttm_bo_wait_ctx(bo, ctx);
+       if (ret)
+               goto out;
+
+       if (old_mem->mem_type == TTM_PL_SYSTEM && bo->ttm == NULL) {
+               ttm_bo_move_null(bo, new_mem);
+               return 0;
+       }
+       ret = ttm_bo_move_memcpy(bo, ctx, new_mem);
+out:
+       if (ret) {
+               swap(*new_mem, bo->mem);
+               qxl_bo_move_notify(bo, false, new_mem);
+               swap(*new_mem, bo->mem);
+       }
+       return ret;
+}
+
 static struct ttm_bo_driver qxl_bo_driver = {
        .ttm_tt_create = &qxl_ttm_tt_create,
        .ttm_tt_bind = &qxl_ttm_backend_bind,
 
        struct ttm_resource *old_mem = &bo->mem;
        int r;
 
+       radeon_bo_move_notify(bo, evict, new_mem);
+
        r = ttm_bo_wait_ctx(bo, ctx);
        if (r)
-               return r;
+               goto fail;
 
        /* Can't move a pinned BO */
        rbo = container_of(bo, struct radeon_bo, tbo);
 memcpy:
                r = ttm_bo_move_memcpy(bo, ctx, new_mem);
                if (r) {
-                       return r;
+                       goto fail;
                }
        }
 
        /* update statistics */
        atomic64_add((u64)bo->num_pages << PAGE_SHIFT, &rdev->num_bytes_moved);
        return 0;
+fail:
+       swap(*new_mem, bo->mem);
+       radeon_bo_move_notify(bo, false, new_mem);
+       swap(*new_mem, bo->mem);
+       return r;
 }
 
 static int radeon_ttm_io_mem_reserve(struct ttm_bo_device *bdev, struct ttm_resource *mem)
 
                }
        }
 
-       if (bdev->driver->move_notify)
-               bdev->driver->move_notify(bo, evict, mem);
-
        ret = bdev->driver->move(bo, evict, ctx, mem);
-       if (ret) {
-               if (bdev->driver->move_notify) {
-                       swap(*mem, bo->mem);
-                       bdev->driver->move_notify(bo, false, mem);
-                       swap(*mem, bo->mem);
-               }
-
+       if (ret)
                goto out_err;
-       }
 
        ctx->bytes_moved += bo->num_pages << PAGE_SHIFT;
        return 0;
 
        struct ttm_resource_manager *new_man = ttm_manager_type(bo->bdev, new_mem->mem_type);
        int ret;
 
+       vmw_move_notify(bo, evict, new_mem);
+
        if (old_man->use_tt && new_man->use_tt) {
                if (bo->mem.mem_type == TTM_PL_SYSTEM) {
                        ttm_bo_assign_mem(bo, new_mem);
                }
                ret = ttm_bo_wait_ctx(bo, ctx);
                if (ret)
-                       return ret;
+                       goto fail;
 
                vmw_ttm_unbind(bo->bdev, bo->ttm);
                ttm_resource_free(bo, &bo->mem);
                ttm_bo_assign_mem(bo, new_mem);
                return 0;
        } else {
-               return ttm_bo_move_memcpy(bo, ctx, new_mem);
+               ret = ttm_bo_move_memcpy(bo, ctx, new_mem);
+               if (ret)
+                       goto fail;
        }
+       return 0;
+fail:
+       swap(*new_mem, bo->mem);
+       vmw_move_notify(bo, false, new_mem);
+       swap(*new_mem, bo->mem);
+       return ret;
 }
 
 struct ttm_bo_driver vmw_bo_driver = {