struct clk *clk;
 
        struct iommu_domain *domain;
-       struct iommu_group *group;
        struct kmem_cache *pt_pool;
 };
 
        return &iommu->iommu;
 }
 
-static struct iommu_group *sun50i_iommu_device_group(struct device *dev)
-{
-       struct sun50i_iommu *iommu = sun50i_iommu_from_dev(dev);
-
-       return iommu_group_ref_get(iommu->group);
-}
-
 static int sun50i_iommu_of_xlate(struct device *dev,
                                 struct of_phandle_args *args)
 {
 static const struct iommu_ops sun50i_iommu_ops = {
        .identity_domain = &sun50i_iommu_identity_domain,
        .pgsize_bitmap  = SZ_4K,
-       .device_group   = sun50i_iommu_device_group,
+       .device_group   = generic_single_device_group,
        .domain_alloc_paging = sun50i_iommu_domain_alloc_paging,
        .of_xlate       = sun50i_iommu_of_xlate,
        .probe_device   = sun50i_iommu_probe_device,
        if (!iommu->pt_pool)
                return -ENOMEM;
 
-       iommu->group = iommu_group_alloc();
-       if (IS_ERR(iommu->group)) {
-               ret = PTR_ERR(iommu->group);
-               goto err_free_cache;
-       }
-
        iommu->base = devm_platform_ioremap_resource(pdev, 0);
        if (IS_ERR(iommu->base)) {
                ret = PTR_ERR(iommu->base);
-               goto err_free_group;
+               goto err_free_cache;
        }
 
        irq = platform_get_irq(pdev, 0);
        if (irq < 0) {
                ret = irq;
-               goto err_free_group;
+               goto err_free_cache;
        }
 
        iommu->clk = devm_clk_get(&pdev->dev, NULL);
        if (IS_ERR(iommu->clk)) {
                dev_err(&pdev->dev, "Couldn't get our clock.\n");
                ret = PTR_ERR(iommu->clk);
-               goto err_free_group;
+               goto err_free_cache;
        }
 
        iommu->reset = devm_reset_control_get(&pdev->dev, NULL);
        if (IS_ERR(iommu->reset)) {
                dev_err(&pdev->dev, "Couldn't get our reset line.\n");
                ret = PTR_ERR(iommu->reset);
-               goto err_free_group;
+               goto err_free_cache;
        }
 
        ret = iommu_device_sysfs_add(&iommu->iommu, &pdev->dev,
                                     NULL, dev_name(&pdev->dev));
        if (ret)
-               goto err_free_group;
+               goto err_free_cache;
 
        ret = iommu_device_register(&iommu->iommu, &sun50i_iommu_ops, &pdev->dev);
        if (ret)
 err_remove_sysfs:
        iommu_device_sysfs_remove(&iommu->iommu);
 
-err_free_group:
-       iommu_group_put(iommu->group);
-
 err_free_cache:
        kmem_cache_destroy(iommu->pt_pool);