void vbox_report_caps(struct vbox_private *vbox);
 
 int vbox_mm_init(struct vbox_private *vbox);
-void vbox_mm_fini(struct vbox_private *vbox);
 
 /* vbox_irq.c */
 int vbox_irq_init(struct vbox_private *vbox);
 
 int vbox_mm_init(struct vbox_private *vbox)
 {
        int ret;
+       resource_size_t base, size;
        struct drm_device *dev = &vbox->ddev;
        struct pci_dev *pdev = to_pci_dev(dev->dev);
 
-       ret = drmm_vram_helper_init(dev, pci_resource_start(pdev, 0),
-                                      vbox->available_vram_size);
+       base = pci_resource_start(pdev, 0);
+       size = pci_resource_len(pdev, 0);
+
+       /* Don't fail on errors, but performance might be reduced. */
+       devm_arch_phys_wc_add(&pdev->dev, base, size);
+
+       ret = drmm_vram_helper_init(dev, base, vbox->available_vram_size);
        if (ret) {
                DRM_ERROR("Error initializing VRAM MM; %d\n", ret);
                return ret;
        }
 
-       vbox->fb_mtrr = arch_phys_wc_add(pci_resource_start(pdev, 0),
-                                        pci_resource_len(pdev, 0));
        return 0;
 }
-
-void vbox_mm_fini(struct vbox_private *vbox)
-{
-       arch_phys_wc_del(vbox->fb_mtrr);
-}