static int msm_iommu_add_device(struct device *dev)
 {
        struct msm_iommu_dev *iommu;
+       struct iommu_group *group;
        unsigned long flags;
        int ret = 0;
 
 
        spin_unlock_irqrestore(&msm_iommu_lock, flags);
 
-       return ret;
+       if (ret)
+               return ret;
+
+       group = iommu_group_get_for_dev(dev);
+       if (IS_ERR(group))
+               return PTR_ERR(group);
+
+       iommu_group_put(group);
+
+       return 0;
 }
 
 static void msm_iommu_remove_device(struct device *dev)
                iommu_device_unlink(&iommu->iommu, dev);
 
        spin_unlock_irqrestore(&msm_iommu_lock, flags);
+
+       iommu_group_remove_device(dev);
 }
 
 static int msm_iommu_attach_dev(struct iommu_domain *domain, struct device *dev)
        .iova_to_phys = msm_iommu_iova_to_phys,
        .add_device = msm_iommu_add_device,
        .remove_device = msm_iommu_remove_device,
+       .device_group = generic_device_group,
        .pgsize_bitmap = MSM_IOMMU_PGSIZES,
        .of_xlate = qcom_iommu_of_xlate,
 };