struct list_head        client;
        spinlock_t              client_lock;    /* for client list */
        struct device           *dev;
+
+       struct iommu_device     iommu;          /* IOMMU Core handle */
 };
 
 struct gart_domain {
                return PTR_ERR(group);
 
        iommu_group_put(group);
+
+       iommu_device_link(&gart_handle->iommu, dev);
+
        return 0;
 }
 
 static void gart_iommu_remove_device(struct device *dev)
 {
        iommu_group_remove_device(dev);
+       iommu_device_unlink(&gart_handle->iommu, dev);
 }
 
 static const struct iommu_ops gart_iommu_ops = {
        struct resource *res, *res_remap;
        void __iomem *gart_regs;
        struct device *dev = &pdev->dev;
+       int ret;
 
        if (gart_handle)
                return -EIO;
                return -ENXIO;
        }
 
+       ret = iommu_device_sysfs_add(&gart->iommu, &pdev->dev, NULL,
+                                    dev_name(&pdev->dev));
+       if (ret) {
+               dev_err(dev, "Failed to register IOMMU in sysfs\n");
+               return ret;
+       }
+
+       iommu_device_set_ops(&gart->iommu, &gart_iommu_ops);
+
+       ret = iommu_device_register(&gart->iommu);
+       if (ret) {
+               dev_err(dev, "Failed to register IOMMU\n");
+               iommu_device_sysfs_remove(&gart->iommu);
+               return ret;
+       }
+
        gart->dev = &pdev->dev;
        spin_lock_init(&gart->pte_lock);
        spin_lock_init(&gart->client_lock);
 {
        struct gart_device *gart = platform_get_drvdata(pdev);
 
+       iommu_device_unregister(&gart->iommu);
+       iommu_device_sysfs_remove(&gart->iommu);
+
        writel(0, gart->regs + GART_CONFIG);
        if (gart->savedata)
                vfree(gart->savedata);