unsigned int dma_mask_size : 8;
        /* Chipset specific GTT setup */
        int (*setup)(void);
+       /* This should undo anything done in ->setup() save the unmapping
+        * of the mmio register file, that's done in the generic code. */
+       void (*cleanup)(void);
        void (*write_entry)(dma_addr_t addr, unsigned int entry, unsigned int flags);
        /* Flags is a more or less chipset specific opaque value.
         * For chipsets that need to support old ums (non-gem) code, this
 
 static void intel_gtt_cleanup(void)
 {
-       if (intel_private.i9xx_flush_page)
-               iounmap(intel_private.i9xx_flush_page);
-       if (intel_private.resource_valid)
-               release_resource(&intel_private.ifp_resource);
-       intel_private.ifp_resource.start = 0;
-       intel_private.resource_valid = 0;
+       intel_private.driver->cleanup();
+
        iounmap(intel_private.gtt);
        iounmap(intel_private.registers);
        
        intel_private.gtt = ioremap(intel_private.gtt_bus_addr,
                                    gtt_map_size);
        if (!intel_private.gtt) {
+               intel_private.driver->cleanup();
                iounmap(intel_private.registers);
                return -ENOMEM;
        }
        /* we have to call this as early as possible after the MMIO base address is known */
        intel_private.base.gtt_stolen_entries = intel_gtt_stolen_entries();
        if (intel_private.base.gtt_stolen_entries == 0) {
+               intel_private.driver->cleanup();
                iounmap(intel_private.registers);
                iounmap(intel_private.gtt);
                return -ENOMEM;
        return 0;
 }
 
-static void intel_i830_fini_flush(void)
+static void i830_cleanup(void)
 {
        kunmap(intel_private.i8xx_page);
        intel_private.i8xx_flush_page = NULL;
 
        intel_private.i8xx_flush_page = kmap(intel_private.i8xx_page);
        if (!intel_private.i8xx_flush_page)
-               intel_i830_fini_flush();
+               i830_cleanup();
 }
 
 /* The chipset_flush interface needs to get data that has already been
                        "can't ioremap flush page - no chipset flushing\n");
 }
 
+static void i9xx_cleanup(void)
+{
+       if (intel_private.i9xx_flush_page)
+               iounmap(intel_private.i9xx_flush_page);
+       if (intel_private.resource_valid)
+               release_resource(&intel_private.ifp_resource);
+       intel_private.ifp_resource.start = 0;
+       intel_private.resource_valid = 0;
+}
+
 static void i9xx_chipset_flush(void)
 {
        if (intel_private.i9xx_flush_page)
        writel(addr | pte_flags, intel_private.gtt + entry);
 }
 
+static void gen6_cleanup(void)
+{
+}
+
 static int i9xx_setup(void)
 {
        u32 reg_addr;
 static const struct intel_gtt_driver i8xx_gtt_driver = {
        .gen = 2,
        .setup = i830_setup,
+       .cleanup = i830_cleanup,
        .write_entry = i830_write_entry,
        .dma_mask_size = 32,
        .check_flags = i830_check_flags,
 static const struct intel_gtt_driver i915_gtt_driver = {
        .gen = 3,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        /* i945 is the last gpu to need phys mem (for overlay and cursors). */
        .write_entry = i830_write_entry, 
        .dma_mask_size = 32,
        .gen = 3,
        .is_g33 = 1,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        .write_entry = i965_write_entry,
        .dma_mask_size = 36,
        .check_flags = i830_check_flags,
        .gen = 3,
        .is_pineview = 1, .is_g33 = 1,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        .write_entry = i965_write_entry,
        .dma_mask_size = 36,
        .check_flags = i830_check_flags,
 static const struct intel_gtt_driver i965_gtt_driver = {
        .gen = 4,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        .write_entry = i965_write_entry,
        .dma_mask_size = 36,
        .check_flags = i830_check_flags,
 static const struct intel_gtt_driver g4x_gtt_driver = {
        .gen = 5,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        .write_entry = i965_write_entry,
        .dma_mask_size = 36,
        .check_flags = i830_check_flags,
        .gen = 5,
        .is_ironlake = 1,
        .setup = i9xx_setup,
+       .cleanup = i9xx_cleanup,
        .write_entry = i965_write_entry,
        .dma_mask_size = 36,
        .check_flags = i830_check_flags,
 static const struct intel_gtt_driver sandybridge_gtt_driver = {
        .gen = 6,
        .setup = i9xx_setup,
+       .cleanup = gen6_cleanup,
        .write_entry = gen6_write_entry,
        .dma_mask_size = 40,
        .check_flags = gen6_check_flags,