/* ---------------------------------------------------------------------- */
 
 /* bochs_hw.c */
-int bochs_hw_init(struct drm_device *dev, uint32_t flags);
+int bochs_hw_init(struct drm_device *dev);
 void bochs_hw_fini(struct drm_device *dev);
 
 void bochs_hw_setmode(struct bochs_device *bochs,
 
        dev->dev_private = NULL;
 }
 
-static int bochs_load(struct drm_device *dev, unsigned long flags)
+static int bochs_load(struct drm_device *dev)
 {
        struct bochs_device *bochs;
        int ret;
        dev->dev_private = bochs;
        bochs->dev = dev;
 
-       ret = bochs_hw_init(dev, flags);
+       ret = bochs_hw_init(dev);
        if (ret)
                goto err;
 
 
 static struct drm_driver bochs_driver = {
        .driver_features        = DRIVER_GEM | DRIVER_MODESET,
-       .load                   = bochs_load,
-       .unload                 = bochs_unload,
        .fops                   = &bochs_fops,
        .name                   = "bochs-drm",
        .desc                   = "bochs dispi vga interface (qemu stdvga)",
 static int bochs_pci_probe(struct pci_dev *pdev,
                           const struct pci_device_id *ent)
 {
+       struct drm_device *dev;
        unsigned long fbsize;
        int ret;
 
        if (ret)
                return ret;
 
-       return drm_get_pci_dev(pdev, ent, &bochs_driver);
+       dev = drm_dev_alloc(&bochs_driver, &pdev->dev);
+       if (IS_ERR(dev))
+               return PTR_ERR(dev);
+
+       dev->pdev = pdev;
+       pci_set_drvdata(pdev, dev);
+
+       ret = bochs_load(dev);
+       if (ret)
+               goto err_free_dev;
+
+       ret = drm_dev_register(dev, 0);
+       if (ret)
+               goto err_unload;
+
+       return ret;
+
+err_unload:
+       bochs_unload(dev);
+err_free_dev:
+       drm_dev_put(dev);
+       return ret;
 }
 
 static void bochs_pci_remove(struct pci_dev *pdev)
 {
        struct drm_device *dev = pci_get_drvdata(pdev);
 
-       drm_put_dev(dev);
+       drm_dev_unregister(dev);
+       bochs_unload(dev);
+       drm_dev_put(dev);
 }
 
 static const struct pci_device_id bochs_pci_tbl[] = {