iommu/rockchip: Fix error handling in init

[sfrench/cifs-2.6.git] / drivers / iommu / rockchip-iommu.c

diff --git a/drivers/iommu/rockchip-iommu.c b/drivers/iommu/rockchip-iommu.c
index bd8580b897e9532533edc88f8389edcb4c939639..0ce5e8a0658cdf737fffeeb1e0e36b73a8c5b4f9 100644 (file)
--- a/drivers/iommu/rockchip-iommu.c
+++ b/drivers/iommu/rockchip-iommu.c
@@ -1133,6 +1133,8 @@ static int rk_iommu_probe(struct platform_device *pdev)
        if (!dma_dev)
                dma_dev = &pdev->dev;
 
+       bus_set_iommu(&platform_bus_type, &rk_iommu_ops);
+
        return 0;
 err_remove_sysfs:
        iommu_device_sysfs_remove(&iommu->iommu);
@@ -1176,19 +1178,6 @@ static struct platform_driver rk_iommu_driver = {
 
 static int __init rk_iommu_init(void)
 {
-       struct device_node *np;
-       int ret;
-
-       np = of_find_matching_node(NULL, rk_iommu_dt_ids);
-       if (!np)
-               return 0;
-
-       of_node_put(np);
-
-       ret = bus_set_iommu(&platform_bus_type, &rk_iommu_ops);
-       if (ret)
-               return ret;
-
        return platform_driver_register(&rk_iommu_driver);
 }
 subsys_initcall(rk_iommu_init);