iommu/tegra: gart: Move bus_set_iommu after probe for multi arch

[~andy/linux] / drivers / iommu / tegra-gart.c

diff --git a/drivers/iommu/tegra-gart.c b/drivers/iommu/tegra-gart.c
index c16e8fc8a4bd11677971140d6c2d20f48e6e3582..4c9db62814ffde1c2f3cc513eda788490cb000f4 100644 (file)
--- a/drivers/iommu/tegra-gart.c
+++ b/drivers/iommu/tegra-gart.c
@@ -398,6 +398,7 @@ static int tegra_gart_probe(struct platform_device *pdev)
        do_gart_setup(gart, NULL);
 
        gart_handle = gart;
+       bus_set_iommu(&platform_bus_type, &gart_iommu_ops);
        return 0;
 
 fail:
@@ -450,7 +451,6 @@ static struct platform_driver tegra_gart_driver = {
 
 static int __devinit tegra_gart_init(void)
 {
-       bus_set_iommu(&platform_bus_type, &gart_iommu_ops);
        return platform_driver_register(&tegra_gart_driver);
 }