#include <drm/drm_mm.h>
 #include <drm/i915_drm.h>
 
+#include "gem/i915_gem_region.h"
 #include "i915_drv.h"
 #include "i915_gem_stolen.h"
 
        return 0;
 }
 
-void i915_gem_cleanup_stolen(struct drm_i915_private *dev_priv)
+static void i915_gem_cleanup_stolen(struct drm_i915_private *dev_priv)
 {
        if (!drm_mm_initialized(&dev_priv->mm.stolen))
                return;
        }
 }
 
-int i915_gem_init_stolen(struct drm_i915_private *dev_priv)
+static int i915_gem_init_stolen(struct drm_i915_private *dev_priv)
 {
        resource_size_t reserved_base, stolen_top;
        resource_size_t reserved_total, reserved_size;
 
        i915_gem_stolen_remove_node(dev_priv, stolen);
        kfree(stolen);
+
+       if (obj->mm.region)
+               i915_gem_object_release_memory_region(obj);
 }
 
 static const struct drm_i915_gem_object_ops i915_gem_object_stolen_ops = {
 };
 
 static struct drm_i915_gem_object *
-_i915_gem_object_create_stolen(struct drm_i915_private *dev_priv,
-                              struct drm_mm_node *stolen)
+__i915_gem_object_create_stolen(struct drm_i915_private *dev_priv,
+                               struct drm_mm_node *stolen,
+                               struct intel_memory_region *mem)
 {
        struct drm_i915_gem_object *obj;
        unsigned int cache_level;
        if (err)
                goto cleanup;
 
+       if (mem)
+               i915_gem_object_init_memory_region(obj, mem, 0);
+
        return obj;
 
 cleanup:
        return ERR_PTR(err);
 }
 
-struct drm_i915_gem_object *
-i915_gem_object_create_stolen(struct drm_i915_private *dev_priv,
-                             resource_size_t size)
+static struct drm_i915_gem_object *
+_i915_gem_object_create_stolen(struct intel_memory_region *mem,
+                              resource_size_t size,
+                              unsigned int flags)
 {
+       struct drm_i915_private *dev_priv = mem->i915;
        struct drm_i915_gem_object *obj;
        struct drm_mm_node *stolen;
        int ret;
                goto err_free;
        }
 
-       obj = _i915_gem_object_create_stolen(dev_priv, stolen);
+       obj = __i915_gem_object_create_stolen(dev_priv, stolen, mem);
        if (IS_ERR(obj))
                goto err_remove;
 
        return obj;
 }
 
+struct drm_i915_gem_object *
+i915_gem_object_create_stolen(struct drm_i915_private *dev_priv,
+                             resource_size_t size)
+{
+       return i915_gem_object_create_region(dev_priv->mm.regions[INTEL_REGION_STOLEN],
+                                            size, I915_BO_ALLOC_CONTIGUOUS);
+}
+
+static int init_stolen(struct intel_memory_region *mem)
+{
+       /*
+        * Initialise stolen early so that we may reserve preallocated
+        * objects for the BIOS to KMS transition.
+        */
+       return i915_gem_init_stolen(mem->i915);
+}
+
+static void release_stolen(struct intel_memory_region *mem)
+{
+       i915_gem_cleanup_stolen(mem->i915);
+}
+
+static const struct intel_memory_region_ops i915_region_stolen_ops = {
+       .init = init_stolen,
+       .release = release_stolen,
+       .create_object = _i915_gem_object_create_stolen,
+};
+
+struct intel_memory_region *i915_gem_stolen_setup(struct drm_i915_private *i915)
+{
+       return intel_memory_region_create(i915,
+                                         intel_graphics_stolen_res.start,
+                                         resource_size(&intel_graphics_stolen_res),
+                                         PAGE_SIZE, 0,
+                                         &i915_region_stolen_ops);
+}
+
 struct drm_i915_gem_object *
 i915_gem_object_create_stolen_for_preallocated(struct drm_i915_private *dev_priv,
                                               resource_size_t stolen_offset,
                return ERR_PTR(ret);
        }
 
-       obj = _i915_gem_object_create_stolen(dev_priv, stolen);
+       obj = __i915_gem_object_create_stolen(dev_priv, stolen, NULL);
        if (IS_ERR(obj)) {
                DRM_DEBUG_DRIVER("failed to allocate stolen object\n");
                i915_gem_stolen_remove_node(dev_priv, stolen);