{
        int ret;
 
-       if (!iommu || IS_ERR(iommu))
-               return -ENODEV;
-
        ret = sysfs_add_link_to_group(&iommu->dev->kobj, "devices",
                                      &link->kobj, dev_name(link));
        if (ret)
 
 void iommu_device_unlink(struct iommu_device *iommu, struct device *link)
 {
-       if (!iommu || IS_ERR(iommu))
-               return;
-
        sysfs_remove_link(&link->kobj, "iommu");
        sysfs_remove_link_from_group(&iommu->dev->kobj, "devices", dev_name(link));
 }
 
                goto err_module_put;
        }
 
+       ret = iommu_device_link(iommu_dev, dev);
+       if (ret)
+               goto err_release;
+
        group = ops->device_group(dev);
        if (WARN_ON_ONCE(group == NULL))
                group = ERR_PTR(-EINVAL);
        if (IS_ERR(group)) {
                ret = PTR_ERR(group);
-               goto err_release;
+               goto err_unlink;
        }
        dev->iommu_group = group;
 
                dev->iommu->attach_deferred = ops->is_attach_deferred(dev);
        return 0;
 
+err_unlink:
+       iommu_device_unlink(iommu_dev, dev);
 err_release:
        if (ops->release_device)
                ops->release_device(dev);
 
        lockdep_assert_held(&group->mutex);
 
+       iommu_device_unlink(dev->iommu->iommu_dev, dev);
+
        /*
         * release_device() must stop using any attached domain on the device.
         * If there are still other devices in the group they are not effected
        iommu_group_put(group);
 
        mutex_unlock(&iommu_probe_device_lock);
-       iommu_device_link(dev->iommu->iommu_dev, dev);
 
        return 0;
 
        if (!dev->iommu || !group)
                return;
 
-       iommu_device_unlink(dev->iommu->iommu_dev, dev);
-
        __iommu_group_remove_device(dev);
 }