#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
 
-#include <linux/amba/bus.h>
 #include <linux/delay.h>
 #include <linux/dma-iommu.h>
 #include <linux/dma-map-ops.h>
 #include <linux/module.h>
 #include <linux/of_platform.h>
 #include <linux/pci.h>
-#include <linux/platform_device.h>
 #include <linux/virtio.h>
 #include <linux/virtio_config.h>
 #include <linux/virtio_ids.h>
 
        iommu_device_register(&viommu->iommu, &viommu_ops, parent_dev);
 
-#ifdef CONFIG_PCI
-       if (pci_bus_type.iommu_ops != &viommu_ops) {
-               ret = bus_set_iommu(&pci_bus_type, &viommu_ops);
-               if (ret)
-                       goto err_unregister;
-       }
-#endif
-#ifdef CONFIG_ARM_AMBA
-       if (amba_bustype.iommu_ops != &viommu_ops) {
-               ret = bus_set_iommu(&amba_bustype, &viommu_ops);
-               if (ret)
-                       goto err_unregister;
-       }
-#endif
-       if (platform_bus_type.iommu_ops != &viommu_ops) {
-               ret = bus_set_iommu(&platform_bus_type, &viommu_ops);
-               if (ret)
-                       goto err_unregister;
-       }
-
        vdev->priv = viommu;
 
        dev_info(dev, "input address: %u bits\n",
 
        return 0;
 
-err_unregister:
-       iommu_device_sysfs_remove(&viommu->iommu);
-       iommu_device_unregister(&viommu->iommu);
 err_free_vqs:
        vdev->config->del_vqs(vdev);