1 files changed, 14 insertions, 5 deletions

diff --git a/drivers/fpga/altera-cvp.c b/drivers/fpga/altera-cvp.c
index 77b04e4b3254..dd4edd8f22ce 100644
--- a/drivers/fpga/altera-cvp.c
+++ b/drivers/fpga/altera-cvp.c
@@ -401,6 +401,7 @@ static int altera_cvp_probe(struct pci_dev *pdev,
                             const struct pci_device_id *dev_id)
 {
         struct altera_cvp_conf *conf;
+        struct fpga_manager *mgr;
         u16 cmd, val;
         int ret;
 
@@ -452,16 +453,24 @@ static int altera_cvp_probe(struct pci_dev *pdev,
         snprintf(conf->mgr_name, sizeof(conf->mgr_name), "%s @%s",
                  ALTERA_CVP_MGR_NAME, pci_name(pdev));
 
-        ret = fpga_mgr_register(&pdev->dev, conf->mgr_name,
-                                &altera_cvp_ops, conf);
-        if (ret)
+        mgr = fpga_mgr_create(&pdev->dev, conf->mgr_name,
+                              &altera_cvp_ops, conf);
+        if (!mgr)
+                return -ENOMEM;
+
+        pci_set_drvdata(pdev, mgr);
+
+        ret = fpga_mgr_register(mgr);
+        if (ret) {
+                fpga_mgr_free(mgr);
                 goto err_unmap;
+        }
 
         ret = driver_create_file(&altera_cvp_driver.driver,
                                  &driver_attr_chkcfg);
         if (ret) {
                 dev_err(&pdev->dev, "Can't create sysfs chkcfg file\n");
-                fpga_mgr_unregister(&pdev->dev);
+                fpga_mgr_unregister(mgr);
                 goto err_unmap;
         }
 
@@ -483,7 +492,7 @@ static void altera_cvp_remove(struct pci_dev *pdev)
         u16 cmd;
 
         driver_remove_file(&altera_cvp_driver.driver, &driver_attr_chkcfg);
-        fpga_mgr_unregister(&pdev->dev);
+        fpga_mgr_unregister(mgr);
         pci_iounmap(pdev, conf->map);
         pci_release_region(pdev, CVP_BAR);
         pci_read_config_word(pdev, PCI_COMMAND, &cmd);