#include <drm/drm_gem_framebuffer_helper.h>
 
 #include "framebuffer.h"
+#include "gem.h"
 #include "gtt.h"
 #include "psb_drv.h"
 #include "psb_intel_drv.h"
        /* Begin by trying to use stolen memory backing */
        backing = psb_gtt_alloc_range(dev, aligned_size, "fb", 1, PAGE_SIZE);
        if (backing) {
+               backing->gem.funcs = &psb_gem_object_funcs;
                drm_gem_private_object_init(dev, &backing->gem, aligned_size);
                return backing;
        }
 
 
 #include "psb_drv.h"
 
-void psb_gem_free_object(struct drm_gem_object *obj)
+static vm_fault_t psb_gem_fault(struct vm_fault *vmf);
+
+static void psb_gem_free_object(struct drm_gem_object *obj)
 {
        struct gtt_range *gtt = container_of(obj, struct gtt_range, gem);
 
        return -EINVAL;
 }
 
+static const struct vm_operations_struct psb_gem_vm_ops = {
+       .fault = psb_gem_fault,
+       .open = drm_gem_vm_open,
+       .close = drm_gem_vm_close,
+};
+
+const struct drm_gem_object_funcs psb_gem_object_funcs = {
+       .free = psb_gem_free_object,
+       .vm_ops = &psb_gem_vm_ops,
+};
+
 /**
  *     psb_gem_create          -       create a mappable object
  *     @file: the DRM file of the client
                dev_err(dev->dev, "no memory for %lld byte GEM object\n", size);
                return -ENOSPC;
        }
+       r->gem.funcs = &psb_gem_object_funcs;
        /* Initialize the extra goodies GEM needs to do all the hard work */
        if (drm_gem_object_init(dev, &r->gem, size) != 0) {
                psb_gtt_free_range(dev, r);
  *     vma->vm_private_data points to the GEM object that is backing this
  *     mapping.
  */
-vm_fault_t psb_gem_fault(struct vm_fault *vmf)
+static vm_fault_t psb_gem_fault(struct vm_fault *vmf)
 {
        struct vm_area_struct *vma = vmf->vma;
        struct drm_gem_object *obj;
 
 #ifndef _GEM_H
 #define _GEM_H
 
+extern const struct drm_gem_object_funcs psb_gem_object_funcs;
+
 extern int psb_gem_create(struct drm_file *file, struct drm_device *dev,
                          u64 size, u32 *handlep, int stolen, u32 align);
+
 #endif
 
        .runtime_idle = psb_runtime_idle,
 };
 
-static const struct vm_operations_struct psb_gem_vm_ops = {
-       .fault = psb_gem_fault,
-       .open = drm_gem_vm_open,
-       .close = drm_gem_vm_close,
-};
-
 static const struct file_operations psb_gem_fops = {
        .owner = THIS_MODULE,
        .open = drm_open,
        .irq_uninstall = psb_irq_uninstall,
        .irq_handler = psb_irq_handler,
 
-       .gem_free_object_unlocked = psb_gem_free_object,
-       .gem_vm_ops = &psb_gem_vm_ops,
-
        .dumb_create = psb_gem_dumb_create,
        .ioctls = psb_ioctls,
        .fops = &psb_gem_fops,
 
 extern const struct drm_connector_funcs psb_intel_lvds_connector_funcs;
 
 /* gem.c */
-extern void psb_gem_free_object(struct drm_gem_object *obj);
 extern int psb_gem_get_aperture(struct drm_device *dev, void *data,
                        struct drm_file *file);
 extern int psb_gem_dumb_create(struct drm_file *file, struct drm_device *dev,
                        struct drm_mode_create_dumb *args);
-extern vm_fault_t psb_gem_fault(struct vm_fault *vmf);
 
 /* psb_device.c */
 extern const struct psb_ops psb_chip_ops;