1 files changed, 19 insertions, 5 deletions

diff --git a/drivers/fpga/altera-hps2fpga.c b/drivers/fpga/altera-hps2fpga.c
index 406d2f10741f..e4d39f0a7572 100644
--- a/drivers/fpga/altera-hps2fpga.c
+++ b/drivers/fpga/altera-hps2fpga.c
@@ -139,6 +139,7 @@ static int alt_fpga_bridge_probe(struct platform_device *pdev)
         struct device *dev = &pdev->dev;
         struct altera_hps2fpga_data *priv;
         const struct of_device_id *of_id;
+        struct fpga_bridge *br;
         u32 enable;
         int ret;
 
@@ -190,11 +191,24 @@ static int alt_fpga_bridge_probe(struct platform_device *pdev)
                 }
         }
 
-        ret = fpga_bridge_register(dev, priv->name, &altera_hps2fpga_br_ops,
-                                   priv);
-err:
+        br = fpga_bridge_create(dev, priv->name, &altera_hps2fpga_br_ops, priv);
+        if (!br) {
+                ret = -ENOMEM;
+                goto err;
+        }
+
+        platform_set_drvdata(pdev, br);
+
+        ret = fpga_bridge_register(br);
         if (ret)
-                clk_disable_unprepare(priv->clk);
+                goto err_free;
+
+        return 0;
+
+err_free:
+        fpga_bridge_free(br);
+err:
+        clk_disable_unprepare(priv->clk);
 
         return ret;
 }
@@ -204,7 +218,7 @@ static int alt_fpga_bridge_remove(struct platform_device *pdev)
         struct fpga_bridge *bridge = platform_get_drvdata(pdev);
         struct altera_hps2fpga_data *priv = bridge->priv;
 
-        fpga_bridge_unregister(&pdev->dev);
+        fpga_bridge_unregister(bridge);
 
         clk_disable_unprepare(priv->clk);