#include <linux/pci.h>
 #include <linux/platform_device.h>
 
+#include <linux/amba/bus.h>
+
 #include "io-pgtable.h"
 
 /* MMIO registers */
        iommu_fwspec_free(dev);
 }
 
+static struct iommu_group *arm_smmu_device_group(struct device *dev)
+{
+       struct iommu_group *group;
+
+       /*
+        * We don't support devices sharing stream IDs other than PCI RID
+        * aliases, since the necessary ID-to-device lookup becomes rather
+        * impractical given a potential sparse 32-bit stream ID space.
+        */
+       if (dev_is_pci(dev))
+               group = pci_device_group(dev);
+       else
+               group = generic_device_group(dev);
+
+       return group;
+}
+
 static int arm_smmu_domain_get_attr(struct iommu_domain *domain,
                                    enum iommu_attr attr, void *data)
 {
 
 static int arm_smmu_of_xlate(struct device *dev, struct of_phandle_args *args)
 {
-       /* We only support PCI, for now */
-       if (!dev_is_pci(dev))
-               return -ENODEV;
-
        return iommu_fwspec_add_ids(dev, args->args, 1);
 }
 
        .iova_to_phys           = arm_smmu_iova_to_phys,
        .add_device             = arm_smmu_add_device,
        .remove_device          = arm_smmu_remove_device,
-       .device_group           = pci_device_group,
+       .device_group           = arm_smmu_device_group,
        .domain_get_attr        = arm_smmu_domain_get_attr,
        .domain_set_attr        = arm_smmu_domain_set_attr,
        .of_xlate               = arm_smmu_of_xlate,
 
        /* And we're up. Go go go! */
        of_iommu_set_ops(dev->of_node, &arm_smmu_ops);
+#ifdef CONFIG_PCI
        pci_request_acs();
-       return bus_set_iommu(&pci_bus_type, &arm_smmu_ops);
+       ret = bus_set_iommu(&pci_bus_type, &arm_smmu_ops);
+       if (ret)
+               return ret;
+#endif
+#ifdef CONFIG_ARM_AMBA
+       ret = bus_set_iommu(&amba_bustype, &arm_smmu_ops);
+       if (ret)
+               return ret;
+#endif
+       return bus_set_iommu(&platform_bus_type, &arm_smmu_ops);
 }
 
 static int arm_smmu_device_remove(struct platform_device *pdev)